WIP: planning with ground-truth dynamics

T-will19 · Oct 8, 2021 · 08132bb · 08132bb
1 parent b665d7a
commit 08132bb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mbrl/diagnostics/control_env.py b/mbrl/diagnostics/control_env.py
@@ -83,7 +83,8 @@ def get_random_trajectory(horizon):
     args = parser.parse_args()
 
     mp.set_start_method("spawn")
-    eval_env = handler__.make_env_from_str(args.env)
+    handler = mbrl.util.create_handler_from_str(args.env)
+    eval_env = handler.make_env_from_str(args.env)
     eval_env.seed(args.seed)
     torch.random.manual_seed(args.seed)
     np.random.seed(args.seed)
@@ -161,7 +162,7 @@ def get_random_trajectory(horizon):
                 frames.append(eval_env.render(mode="rgb_array"))
             start = time.time()
 
-            current_state__ = handler__.get_current_state(
+            current_state__ = handler.get_current_state(
                 cast(gym.wrappers.TimeLimit, eval_env)
             )