Remove the pettingzoo and pybullet dependencies

coltra/envs/__init__.py

@@ -2,7 +2,6 @@
 from .base_env import MultiAgentEnv
 from .subproc_vec_env import SubprocVecEnv
 from .probe_envs import probe_env_classes
-from .gym_envs import MultiGymEnv, import_bullet
+from .gym_envs import MultiGymEnv
 from .base_env import Observation, Action
 from .smartnav_envs import SmartNavEnv
-from .pettingzoo_envs import PettingZooEnv

coltra/envs/gym_envs.py

@@ -10,11 +10,6 @@
 from coltra.envs.subproc_vec_env import VecEnv, SubprocVecEnv
 
 
-def import_bullet():
-    # noinspection PyUnresolvedReferences
-    import pybullet_envs
-
-
 class MultiGymEnv(MultiAgentEnv):
     """
     A wrapper for environments that can be `gym.make`'d
@@ -32,8 +27,6 @@ def __init__(
         super().__init__(seed)
         if wrappers is None:
             wrappers = []
-        if "Bullet" in env_name:
-            import_fn = import_bullet
 
         import_fn()
 

requirements.txt

@@ -17,7 +17,6 @@ ipykernel>=5.4.3
 numba>=0.54.0
 pytest>=6.2.2
 coverage>=5.5.0
-pybullet>=3.1.9
 opencv-python~=3.4.15.55
 cloudpickle~=2.0.0
 pillow~=8.4.0

scripts/enjoy_gym.py

@@ -12,8 +12,6 @@
 from coltra.trainers import PPOCrowdTrainer
 from coltra.envs import MultiGymEnv
 
-import pybullet_envs
-
 
 class Parser(BaseParser):
     path: str

scripts/train_gym.py

@@ -11,8 +11,6 @@
 from coltra.trainers import PPOCrowdTrainer
 from coltra.envs import MultiGymEnv
 
-import pybullet_envs
-
 from coltra.wrappers import ObsVecNormWrapper, LastRewardWrapper
 from coltra.wrappers.agent_wrappers import RetNormWrapper