relax hap denpendencies on torch

Signed-off-by: Maroun Touma <[email protected]>
IBM · Nov 25, 2024 · f61493c · f61493c
1 parent 5a018e6
commit f61493c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/transforms/pyproject.toml b/transforms/pyproject.toml
@@ -80,11 +80,11 @@ web2parquet = { file = ["universal/web2parquet/requirements.txt"]}
 # Does not seem to work for our custom layout
 # copy all files to a single src and let automatic discovery find them
 
-[tool.setuptools.package-data]
-"*" = ["*.txt"]
+#[tool.setuptools.package-data]
+#"*" = ["*.txt"]
 
-[tool.setuptools.packages.find]
-where = ["src"]
+#[tool.setuptools.packages.find]
+#where = ["src"]
 
 #[tool.setuptools.package-dir]
 #dpk_web2parquet = "universal/web2parquet/dpk_web2parquet"

diff --git a/transforms/universal/hap/python/requirements.txt b/transforms/universal/hap/python/requirements.txt
@@ -1,5 +1,5 @@
 data-prep-toolkit==0.2.2.dev2
 nltk==3.9.1
 transformers==4.38.2
-torch==2.4.1
+torch>=2.2.2,<=2.4.1
 pandas==2.2.2
diff --git a/transforms/universal/hap/ray/requirements.txt b/transforms/universal/hap/ray/requirements.txt
@@ -2,5 +2,5 @@ data-prep-toolkit[ray]==0.2.2.dev2
 dpk-hap-transform-python==0.2.2.dev2
 nltk==3.9.1
 transformers==4.38.2
-torch==2.4.1
+torch>=2.2.2,<=2.4.1
 pandas==2.2.2