[BUGFIX] Pinning torch nightly to January 13, 2024 to avoid Attribute…

…Error (#3885)
ludwig-ai · Jan 16, 2024 · 31a3957 · 31a3957
1 parent 8e8f47b
commit 31a3957
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
@@ -126,7 +126,8 @@ jobs:
 
           if [ "$PYTORCH" == "nightly" ]; then
             extra_index_url=https://download.pytorch.org/whl/nightly/cpu
-            pip install --pre torch torchdata torchtext torchvision torchaudio --index-url $extra_index_url
+            # This is being tracked as the Issue "https://github.com/ludwig-ai/ludwig/issues/3886" in Ludwig GitHub repository.
+            pip install --pre torch==2.3.0.dev20240111+cpu torchtext torchvision torchaudio --extra-index-url $extra_index_url
 
           else
             extra_index_url=https://download.pytorch.org/whl/cpu