minor fix

Author: iclavera

rand_param_envs/__init__.py

@@ -9,4 +9,11 @@
 register(
     id='HopperRandParams-v0',
     entry_point='rand_param_envs.hopper_rand_params:HopperRandParamsEnv',
-)
+)
+
+register(
+    id='PR2Env-v0',
+    entry_point='rand_param_envs.pr2_env_reach:PR2Env',
+)
+
+

rand_param_envs/assets/meshes/base.stl

@@ -9,7 +9,7 @@ class PR2Env(RandomEnv, utils.EzPickle):
 
     def __init__(self, log_scale_limit=1.):
         self.viewer = None
-        RandomEnv.__init__(self, log_scale_limit, self.FILE, 4)
+        RandomEnv.__init__(self, log_scale_limit, 'pr2.xml', 4)
         utils.EzPickle.__init__(self)
 
     def _get_obs(self):
@@ -79,4 +79,4 @@ def viewer_setup(self):
         print(env.model.body_mass)
         for _ in range(100):
             env.render()
-            env.step(env.action_space.sample())
+            env.step(env.action_space.sample())