limit torch version to <2.3 for now. Inconsistent expected results in…

… unit tests
InterDigitalInc · May 1, 2024 · bbc4a76 · bbc4a76
1 parent 065f018
commit bbc4a76
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -140,7 +140,7 @@ def get_extra_requirements():
         "pandas",
         "scipy",
         "matplotlib",
-        "torch>=1.7.1",
+        "torch>=1.7.1, <2.3",
         "torch-geometric>=2.3.0",
         "typing-extensions>=4.0.0",
         "torchvision",