| | import os |
| | import sys |
| | from setuptools import setup, find_packages |
| | from setuptools.command.install import install |
| | import mlagents_envs |
| |
|
| | VERSION = mlagents_envs.__version__ |
| | EXPECTED_TAG = mlagents_envs.__release_tag__ |
| |
|
| | here = os.path.abspath(os.path.dirname(__file__)) |
| |
|
| |
|
| | class VerifyVersionCommand(install): |
| | """ |
| | Custom command to verify that the git tag is the expected one for the release. |
| | Originally based on https://circleci.com/blog/continuously-deploying-python-packages-to-pypi-with-circleci/ |
| | This differs slightly because our tags and versions are different. |
| | """ |
| |
|
| | description = "verify that the git tag matches our version" |
| |
|
| | def run(self): |
| | tag = os.getenv("GITHUB_REF", "NO GITHUB TAG!").replace("refs/tags/", "") |
| |
|
| | if tag != EXPECTED_TAG: |
| | info = "Git tag: {} does not match the expected tag of this app: {}".format( |
| | tag, EXPECTED_TAG |
| | ) |
| | sys.exit(info) |
| |
|
| |
|
| | setup( |
| | name="mlagents_envs", |
| | version=VERSION, |
| | description="Unity Machine Learning Agents Interface", |
| | url="https://github.com/Unity-Technologies/ml-agents", |
| | author="Unity Technologies", |
| | author_email="ML-Agents@unity3d.com", |
| | classifiers=[ |
| | "Intended Audience :: Developers", |
| | "Topic :: Scientific/Engineering :: Artificial Intelligence", |
| | "License :: OSI Approved :: Apache Software License", |
| | "Programming Language :: Python :: 3.8", |
| | "Programming Language :: Python :: 3.9", |
| | "Programming Language :: Python :: 3.10", |
| | ], |
| | packages=find_packages( |
| | exclude=["*.tests", "*.tests.*", "tests.*", "tests", "colabs", "*.ipynb"] |
| | ), |
| | zip_safe=False, |
| | install_requires=[ |
| | "cloudpickle", |
| | "grpcio>=1.11.0", |
| | "Pillow>=4.2.1", |
| | "protobuf>=3.6,<3.20", |
| | "pyyaml>=3.1.0", |
| | "gym>=0.21.0", |
| | "pettingzoo==1.15.0", |
| | "numpy==1.21.2", |
| | "filelock>=3.4.0", |
| | ], |
| | python_requires=">=3.8.13,<=3.10.12", |
| | |
| | cmdclass={"verify": VerifyVersionCommand}, |
| | ) |
| |
|
