Remove imports

stepjam · Jun 13, 2024 · c70fef4 · c70fef4
1 parent 9f14cd3
commit c70fef4
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 8 deletions.
diff --git a/examples/rlbench_gym.py b/examples/rlbench_gym.py
@@ -1,6 +1,4 @@
-import time
 import gymnasium as gym
-import rlbench.gym
 from gymnasium.utils.performance import benchmark_step
 
 env = gym.make('rlbench/reach_target-vision-v0', render_mode="rgb_array")

diff --git a/rlbench/gym/rlbench_env.py b/rlbench/gym/rlbench_env.py
@@ -104,6 +104,7 @@ def reset(self, seed=None, options=None):
         np.random.seed(seed=seed)
         reset_to_demo = None
         if options is not None:
+            # TODO: Write test for this
             reset_to_demo = options.get("reset_to_demo", None)
 
         if reset_to_demo is None:

diff --git a/tests/unit/test_gym.py b/tests/unit/test_gym.py
@@ -1,13 +1,7 @@
 import numpy as np
 import gymnasium as gym
-import rlbench.gym
 import pytest
-from gymnasium.utils.passive_env_checker import check_obs
 from gymnasium.utils.env_checker import check_env
-import threading
-
-import inspect
-from copy import deepcopy
 
 import numpy as np