From 1e62da0833af963f188dc86e9a10fcb61528ec36 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Thu, 18 May 2023 18:03:42 +0200 Subject: [PATCH] Use shimmy as a binding for dmc, also 'testing' seperate from 'all' pip install .[all] will no longer install components only required for testing pip install .[testing] will also install all compionents required to run all tests --- setup.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index 7afd526..c477a8a 100644 --- a/setup.py +++ b/setup.py @@ -6,13 +6,12 @@ from setuptools import setup, find_packages # Environment-specific dependencies for dmc and metaworld extras = { - 'dmc': ['dm_control>=1.0.1'], + 'dmc': ['dm_control>=1.0.1', 'shimmy[dm-control]'], 'metaworld': ['metaworld @ git+https://github.com/Farama-Foundation/Metaworld.git@3ced29c8cee6445386eba32e92870d664ad5e6e3#egg=metaworld', 'mujoco-py<2.2,>=2.1', ], 'box2d': ['gymnasium[box2d]>=0.26.0'], - 'testing': ['pytest'], - "mujoco": ["gymnasium[mujoco]"], + 'mujoco': ['gymnasium[mujoco]>0.26.0'], } # All dependencies @@ -20,6 +19,8 @@ all_groups = set(extras.keys()) extras["all"] = list(set(itertools.chain.from_iterable( map(lambda group: extras[group], all_groups)))) +extras['testing'] = extras["all"] + ['pytest'] + def find_package_data(extensions_to_include: List[str]) -> List[str]: envs_dir = Path("fancy_gym/envs/mujoco")