From c70fef489ddbc2dd6877b1e8609d347696329b34 Mon Sep 17 00:00:00 2001
From: Eugene Teoh <50418168+eugeneteoh@users.noreply.github.com>
Date: Thu, 13 Jun 2024 20:35:25 +0100
Subject: [PATCH] Remove imports

---
 examples/rlbench_gym.py    | 2 --
 rlbench/gym/rlbench_env.py | 1 +
 tests/unit/test_gym.py     | 6 ------
 3 files changed, 1 insertion(+), 8 deletions(-)

diff --git a/examples/rlbench_gym.py b/examples/rlbench_gym.py
index 5f12bb881..803415b72 100644
--- 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
index b86c6c883..a36957ad9 100644
--- 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
index cbc9ea153..6cf00b364 100644
--- 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