That was definitely the best episode they've had this season, even if it was a two parter.
This was basically "The Winter Solstice" for this season. It's actually exactly the same thing. Korra was shown the past, much like Aang was, that told her what she really needed to do.
Happy the amnesia storyline actually ended quickly and was used for a good purpose.