thin wrapper around pre-trained roboschool agents

hw1/roboschool_agents.py

+import RoboschoolAnt_v1_2017jul
+import RoboschoolAtlasForwardWalk_v1_2017jul
+import RoboschoolHumanoid_v1_2017jul
+
+
+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)
+    if 'Atlas' in env_name:
+        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)
+
+    return pi