Update test.py

test.py

@@ -393,11 +393,11 @@ def train_pong():
 
 if __name__ == "__main__":
     # Train CartPole
-    cartpole_policy, cartpole_train_rewards, cartpole_eval_rewards = train_cartpole()
+    #cartpole_policy, cartpole_train_rewards, cartpole_eval_rewards = train_cartpole()
     
     # Train Pong (this will take longer)
-    print("\n\nNote: Pong training will take significantly longer (may take hours)")
-    print("You may want to reduce num_episodes if just testing the code.\n")
+    #print("\n\nNote: Pong training will take significantly longer (may take hours)")
+    #print("You may want to reduce num_episodes if just testing the code.\n")
     
     # Uncomment the line below to train Pong
-    # pong_policy, pong_train_rewards, pong_eval_rewards = train_pong()
+    pong_policy, pong_train_rewards, pong_eval_rewards = train_pong()