From f61493cf1915638e0a4ff4f94c824b02c69833a7 Mon Sep 17 00:00:00 2001 From: Maroun Touma Date: Mon, 25 Nov 2024 17:53:00 -0500 Subject: [PATCH] relax hap denpendencies on torch Signed-off-by: Maroun Touma --- transforms/pyproject.toml | 8 ++++---- transforms/universal/hap/python/requirements.txt | 2 +- transforms/universal/hap/ray/requirements.txt | 2 +- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/transforms/pyproject.toml b/transforms/pyproject.toml index 2357553e4..3c1f64c32 100644 --- 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 index 505dd9ceb..ba8948477 100644 --- 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 index 0ed65f625..34e1d6932 100644 --- 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