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
This commit is contained in:
Dominik Moritz Roth 2023-05-18 18:03:42 +02:00
parent a559f92562
commit 1e62da0833

View File

@ -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")