pepify

hw1/roboschool_agents.py

@@ -7,10 +7,13 @@ def load_policy(env_name, env):
     pi = None
 
     if 'Ant' in env_name:
-        pi = RoboschoolAnt_v1_2017jul.ZooPolicyTensorflow("mymodel1", env.observation_space, env.action_space)
+        pi = RoboschoolAnt_v1_2017jul.ZooPolicyTensorflow(
+            "mymodel1", env.observation_space, env.action_space)
     if 'Atlas' in env_name:
-        pi = RoboschoolAtlasForwardWalk_v1_2017jul.ZooPolicyTensorflow("mymodel1", env.observation_space, env.action_space)
+        pi = RoboschoolAtlasForwardWalk_v1_2017jul.ZooPolicyTensorflow(
+            "mymodel1", env.observation_space, env.action_space)
     if 'Human' in env_name:
-        pi = RoboschoolHumanoid_v1_2017jul.ZooPolicyTensorflow("mymodel1", env.observation_space, env.action_space)
+        pi = RoboschoolHumanoid_v1_2017jul.ZooPolicyTensorflow(
+            "mymodel1", env.observation_space, env.action_space)
 
     return pi