Update test2.py

test2.py

@@ -446,5 +446,4 @@ if __name__ == "__main__":
     #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()