[CI] Remove gym import in test_libs.py

ghstack-source-id: b047458
Pull Request resolved: #2719

Author: Vincent Moens

test/test_libs.py

@@ -7,8 +7,6 @@
 import importlib.util
 import urllib.error
 
-from gym.core import ObsType
-
 _has_isaac = importlib.util.find_spec("isaacgym") is not None
 
 if _has_isaac:
@@ -25,7 +23,7 @@
 from contextlib import nullcontext
 from pathlib import Path
 from sys import platform
-from typing import Optional, Tuple, Union
+from typing import Optional, Union
 from unittest import mock
 
 import numpy as np
@@ -653,14 +651,17 @@ def __init__(self):
                 self.observation_space = gym.spaces.Box(-1, 1)
 
             def step(self, action):
+                assert isinstance(action, dict)
+                assert isinstance(action["a0"], np.ndarray)
+                assert isinstance(action["a1"], np.ndarray)
                 return (0.5, 0.0, False, False, {})
 
             def reset(
                 self,
                 *,
                 seed: Optional[int] = None,
                 options: Optional[dict] = None,
-            ) -> Tuple[ObsType, dict]:
+            ):
                 return (0.0, {})
 
         env = CompositeActionEnv()
@@ -686,14 +687,17 @@ def __init__(self):
                 self.observation_space = gym.spaces.Box(-1, 1)
 
             def step(self, action):
+                assert isinstance(action, dict)
+                assert isinstance(action["a0"], np.ndarray)
+                assert isinstance(action["a1"], np.ndarray)
                 return (0.5, 0.0, False, False, {})
 
             def reset(
                 self,
                 *,
                 seed: Optional[int] = None,
                 options: Optional[dict] = None,
-            ) -> Tuple[ObsType, dict]:
+            ):
                 return (0.0, {})
 
         env = CompositeActionEnv()