diff --git a/setup.py b/setup.py index 6cfb14fb..c080aed7 100644 --- a/setup.py +++ b/setup.py @@ -55,6 +55,8 @@ def get_version(rel_path): "pettingzoo == 1.24.0", "gymnasium == 0.29.0", "stable-baselines3 == 2.1.0", + "transformers >= 4.33.1", + "scipy >= 1.11.2", ], packages=find_packages(), include_package_data=True,