fix pr requests

interTwin-eu · Sep 18, 2024 · d961502 · d961502
1 parent 9651e1f
commit d961502
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 13 deletions.
diff --git a/env-files/torch/generic_torch.sh b/env-files/torch/generic_torch.sh
@@ -56,16 +56,16 @@ if [ -f "${cDir}/$ENV_NAME/bin/torchrun" ]; then
 else
   if [ -z "$NO_CUDA" ] ; then
     pip install --no-cache-dir \
-    'torch==2.4.0' torchvision torchaudio --index-url "$PIP_INDEX_TORCH_CUDA" || exit 1
+    'torch==2.4.*' torchvision torchaudio --index-url "$PIP_INDEX_TORCH_CUDA" || exit 1
   else
     # CPU only installation for MacOS
     if [[ "$OSTYPE" =~ ^darwin ]] ; then
       pip install --no-cache-dir \
-        'torch==2.4.0' torchvision torchaudio || exit 1
+        'torch==2.4.*' torchvision torchaudio || exit 1
     else
     # CPU only installation for other OSs
       pip install --no-cache-dir \
-         'torch==2.4.0' torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu || exit 1
+         'torch==2.4.*' torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu || exit 1
     fi
   fi
 fi

diff --git a/pyproject.toml b/pyproject.toml
@@ -52,7 +52,7 @@ dependencies = [
 
 # Torch should be already installed, but it is specified here to
 # prevent itwinai dependencies from cheing its version.
-# torch = ["torch==2.1.*", "lightning==2.*", "torchmetrics"]
+torch = ["lightning==2.*", "torchmetrics"]
 # torch-cpu = [
 #     "torch==2.1.*",
 #     "torchvision",

diff --git a/tests/use-cases/test_3dgan.py b/tests/use-cases/test_3dgan.py
@@ -92,12 +92,3 @@ def test_3dgan_inference(
 
         # Running the inference
         subprocess.run(run_inference_cmd.split(), check=True, cwd=temp_dir)
-
-
-
-
-
-
-
-
-
Original file line number	Diff line number	Diff line change
Expand Up		@@ -92,12 +92,3 @@ def test_3dgan_inference(

		# Running the inference
		subprocess.run(run_inference_cmd.split(), check=True, cwd=temp_dir)