Add torchdata as a dependency for nightly installation

ludwig-ai · Dec 6, 2023 · 1651fb5 · 1651fb5
1 parent 0de337c
commit 1651fb5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
@@ -29,7 +29,7 @@ jobs:
             pytorch-version: 2.0.0
             torchscript-version: 1.10.2
             ray-version: 2.2.0
-          - python-version: 3.9
+          - python-version: "3.9"
             pytorch-version: 2.1.1
             torchscript-version: 1.10.2
             ray-version: 2.3.1
@@ -126,7 +126,9 @@ jobs:
 
           if [ "$PYTORCH" == "nightly" ]; then
             extra_index_url=https://download.pytorch.org/whl/nightly/cpu
-            pip install --pre torch torchtext torchvision torchaudio --index-url $extra_index_url
+            # Temporary Fix - remove torchdata once https://github.com/pytorch/text/pull/2218 lands.
+            pip install torchdata --extra-index-url https://download.pytorch.org/whl/cpu
+            pip install --pre torch torchdata torchtext torchvision torchaudio --index-url $extra_index_url
 
           else
             extra_index_url=https://download.pytorch.org/whl/cpu