HW5c fix (nit): make `sample_trajectory() part of Problem 1

not how it should be since HW2 sampling should work here
the proper fix will wait until next time around

Author: katerakelly

hw5/meta/train_policy.py

@@ -358,7 +358,7 @@ def sample_trajectory(self, env, animate_this_episode, is_evaluation):
         env.reset_task(is_evaluation=is_evaluation)
         stats = []
         #====================================================================================#
-        #                           ----------PROBLEM 2----------
+        #                           ----------PROBLEM 1----------
         #====================================================================================#
         ep_steps = 0
         steps = 0