infra: update torch constraints for testing (#258)

amazon-braket · Jun 7, 2024 · 7cfecb8 · 7cfecb8
1 parent 1c64c85
commit 7cfecb8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -70,7 +70,7 @@
             "sphinx-rtd-theme",
             "sphinxcontrib-apidoc",
             "tox",
-            "torch>=1.11",
+            "torch>2",
             TF_VERSION,
         ]
     },