Upper bound requirements (#222)

* ci: upper-bound torchmetrics * ci: upper bound accelerate and sklearn
GT4SD · Jul 12, 2023 · ea9c299 · ea9c299
1 parent bff9e2f
commit ea9c299
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 # pypi requirements
-accelerate>=0.12
+accelerate>=0.12,<0.20.0
 datasets>=1.11.0
 diffusers<=0.6.0
 importlib-metadata>=1.7.0,<5.0.0 # temporary: https://github.com/python/importlib_metadata/issues/409
@@ -28,7 +28,7 @@ rdkit>=2022.3.5
 regex>=2.5.91
 reinvent-chemistry==0.0.38
 sacremoses>=0.0.41
-scikit-learn>=1.0.0
+scikit-learn>=1.0.0,<1.3.0
 scikit-optimize>=0.8.1
 scipy>=1.0.0
 sentencepiece>=0.1.95
@@ -40,7 +40,7 @@ tensorboard-data-server<=0.6.1
 tensorflow>=2.1.0,<2.11.0
 tensorflow-io-gcs-filesystem<0.32.0
 torchdrug>=0.2.0
-torchmetrics>=0.7.0
+torchmetrics>=0.7.0,<1.0.0
 transformers>=4.22.0,<=4.24.0
 typing_extensions>=3.7.4.3
 wheel>=0.26