Provide setup.py as fallback (again)
This commit is contained in:
parent
5e63a72882
commit
b183c002ef
69
setup.py
Normal file
69
setup.py
Normal file
@ -0,0 +1,69 @@
|
||||
# We still provide a setup.py for backwards compatability.
|
||||
# But the pyproject.toml should be prefered.
|
||||
import itertools
|
||||
from pathlib import Path
|
||||
from typing import List
|
||||
|
||||
from setuptools import setup, find_packages
|
||||
|
||||
# Environment-specific dependencies for dmc and metaworld
|
||||
extras = {
|
||||
'dmc': ['shimmy[dm-control]', 'Shimmy==1.0.0'],
|
||||
'box2d': ['gymnasium[box2d]>=0.26.0'],
|
||||
'mujoco-legacy': ['mujoco-py >=2.1,<2.2', 'cython<3'],
|
||||
'jax': ["jax >=0.4.0", "jaxlib >=0.4.0"],
|
||||
}
|
||||
|
||||
# All dependencies
|
||||
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")
|
||||
package_data_paths = []
|
||||
for extension in extensions_to_include:
|
||||
package_data_paths.extend([str(path)[10:]
|
||||
for path in envs_dir.rglob(extension)])
|
||||
|
||||
return package_data_paths
|
||||
|
||||
|
||||
setup(
|
||||
author='Fabian Otto, Onur Celik, Dominik Roth, Hongyi Zhou',
|
||||
name='fancy_gym',
|
||||
version='1.0',
|
||||
classifiers=[
|
||||
'Development Status :: 4 - Beta',
|
||||
'Intended Audience :: Science/Research',
|
||||
'License :: OSI Approved :: MIT License',
|
||||
'Natural Language :: English',
|
||||
'Operating System :: OS Independent',
|
||||
'Topic :: Scientific/Engineering :: Artificial Intelligence',
|
||||
'Programming Language :: Python :: 3',
|
||||
'Programming Language :: Python :: 3.7',
|
||||
'Programming Language :: Python :: 3.8',
|
||||
'Programming Language :: Python :: 3.9',
|
||||
'Programming Language :: Python :: 3.10',
|
||||
'Programming Language :: Python :: 3.11',
|
||||
],
|
||||
extras_require=extras,
|
||||
install_requires=[
|
||||
'mp_pytorch<=0.1.3',
|
||||
'mujoco==2.3.3',
|
||||
'gymnasium[mujoco]>=0.26.0'
|
||||
],
|
||||
packages=[package for package in find_packages(
|
||||
) if package.startswith("fancy_gym")],
|
||||
package_data={
|
||||
"fancy_gym": find_package_data(extensions_to_include=["*.stl", "*.xml"])
|
||||
},
|
||||
python_requires=">=3.7",
|
||||
url='https://github.com/ALRhub/fancy_gym/',
|
||||
license='MIT license',
|
||||
author_email='',
|
||||
description='Fancy Gym: Unifying interface for various RL benchmarks with support for Black Box approaches.'
|
||||
)
|
Loading…
Reference in New Issue
Block a user