您最多选择25个主题
主题必须以中文或者字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
43 行
1.3 KiB
43 行
1.3 KiB
#!/usr/bin/env python
|
|
|
|
import os
|
|
import sys
|
|
from setuptools import setup, find_packages
|
|
from setuptools.command.install import install
|
|
import gym_unity
|
|
|
|
VERSION = gym_unity.__version__
|
|
EXPECTED_TAG = gym_unity.__release_tag__
|
|
|
|
|
|
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="gym_unity",
|
|
version=VERSION,
|
|
description="Unity Machine Learning Agents Gym Interface",
|
|
license="Apache License 2.0",
|
|
author="Unity Technologies",
|
|
author_email="ML-Agents@unity3d.com",
|
|
url="https://github.com/Unity-Technologies/ml-agents",
|
|
packages=find_packages(),
|
|
install_requires=["gym", f"mlagents_envs=={VERSION}"],
|
|
cmdclass={"verify": VerifyVersionCommand},
|
|
)
|