Merge pull request #386 from ntumlgroup/dependency

Fix Autotest: Set Pytorch <=2.3 to resolve torchtext dependency issue
ASUS-AICS · Aug 22, 2024 · 91e6a4a · 91e6a4a
2 parents eb711ba + 849a8eb
commit 91e6a4a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/requirements_nn.txt b/requirements_nn.txt
@@ -1,7 +1,8 @@
 nltk
 # wait for https://github.com/Lightning-AI/pytorch-lightning/pull/19191
 lightning==2.0.9
-torch
+# https://github.com/pytorch/text/releases
+torch<=2.3
 torchmetrics==0.10.3
 torchtext
 transformers
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = libmultilabel
-version = 0.7.0
+version = 0.7.1
 author = LibMultiLabel Team
 license = MIT License
 license_file = LICENSE
@@ -39,7 +39,7 @@ python_requires = >=3.8
 nn =
     lightning==2.0.9
     nltk
-    torch
+    torch<=2.3
     torchmetrics==0.10.3
     torchtext
     transformers