From dd0baf7f3c4ff7e9f0b4e61fa1401e9039453183 Mon Sep 17 00:00:00 2001
From: Eugene Teoh <50418168+eugeneteoh@users.noreply.github.com>
Date: Thu, 13 Jun 2024 20:50:49 +0100
Subject: [PATCH] Add import

---
 tests/unit/test_gym.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/tests/unit/test_gym.py b/tests/unit/test_gym.py
index 6cf00b364..f5e4a87d3 100644
--- a/tests/unit/test_gym.py
+++ b/tests/unit/test_gym.py
@@ -2,10 +2,7 @@
 import gymnasium as gym
 import pytest
 from gymnasium.utils.env_checker import check_env
-
-import numpy as np
-
-import gymnasium as gym
+import rlbench.gym
 
 @pytest.mark.parametrize("env_id", ['rlbench/reach_target-state-v0', ])
 def test_env(env_id):
@@ -22,5 +19,4 @@ def test_env(env_id):
         assert np.allclose(obs["joint_positions"], obs2["joint_positions"], atol=1e-3)
         assert np.allclose(obs["task_low_dim_state"], obs2["task_low_dim_state"])
 
-    breakpoint()
     env.close()