dodox/fancy_gym
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use shimmy as a binding for dmc, also 'testing' seperate from 'all'

Browse Source 
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
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
setup.py
View File

@ -6,13 +6,12 @@ from setuptools import setup, find_packages
# Environment-specific dependencies for dmc and metaworld # Environment-specific dependencies for dmc and metaworld
extras = { 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', 'metaworld': ['metaworld @ git+https://github.com/Farama-Foundation/Metaworld.git@3ced29c8cee6445386eba32e92870d664ad5e6e3#egg=metaworld',
'mujoco-py<2.2,>=2.1', 'mujoco-py<2.2,>=2.1',
], ],
'box2d': ['gymnasium[box2d]>=0.26.0'], 'box2d': ['gymnasium[box2d]>=0.26.0'],
'testing': ['pytest'], 'mujoco': ['gymnasium[mujoco]>0.26.0'],
"mujoco": ["gymnasium[mujoco]"],
} }
# All dependencies # All dependencies
@ -20,6 +19,8 @@ all_groups = set(extras.keys())
extras["all"] = list(set(itertools.chain.from_iterable( extras["all"] = list(set(itertools.chain.from_iterable(
map(lambda group: extras[group], all_groups)))) map(lambda group: extras[group], all_groups))))
extras['testing'] = extras["all"] + ['pytest']
def find_package_data(extensions_to_include: List[str]) -> List[str]: def find_package_data(extensions_to_include: List[str]) -> List[str]:
envs_dir = Path("fancy_gym/envs/mujoco") envs_dir = Path("fancy_gym/envs/mujoco")