diff --git a/ml-agents-envs/setup.py b/ml-agents-envs/setup.py index a050f4a408..0e0cc57382 100644 --- a/ml-agents-envs/setup.py +++ b/ml-agents-envs/setup.py @@ -51,16 +51,15 @@ def run(self): install_requires=[ "cloudpickle", "grpcio>=1.11.0", - "numpy>=1.14.1", + "numpy>=1.14.1,<1.24", "Pillow>=4.2.1", "protobuf>=3.6", "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.8", + python_requires=">=3.8.13,<3.11.0", # TODO: Remove this once mypy stops having spurious setuptools issues. cmdclass={"verify": VerifyVersionCommand}, # type: ignore ) diff --git a/ml-agents/setup.py b/ml-agents/setup.py index 63bdb78a93..de494c6bd5 100644 --- a/ml-agents/setup.py +++ b/ml-agents/setup.py @@ -68,7 +68,7 @@ def run(self): # https://github.com/Unity-Technologies/ml-agents/blob/release_20_docs/docs/Installation.md#windows-installing-pytorch # Torch only working on python 3.9 for 1.8.0 and above. Details see: # https://github.com/pytorch/pytorch/issues/50014 - "torch>=1.8.0,<=1.11.0;(platform_system!='Windows' and python_version>='3.9')", + "torch>=1.8.0,<=2.0;(platform_system!='Windows' and python_version>='3.9')", "torch>=1.6.0,<1.9.0;(platform_system!='Windows' and python_version<'3.9')", "tensorboard>=1.15", # cattrs 1.1.0 dropped support for python 3.6, but 1.0.0 doesn't work for python 3.9 @@ -79,7 +79,7 @@ def run(self): 'pypiwin32==223;platform_system=="Windows"', "importlib_metadata==4.4; python_version<'3.8'", ], - python_requires=">=3.8.13,<=3.10.8", + python_requires=">=3.8.13,<3.11.0", entry_points={ "console_scripts": [ "mlagents-learn=mlagents.trainers.learn:main",