Disable horovod installation and tests for torch nightly.

ludwig-ai · Jul 14, 2023 · 01a02f6 · 01a02f6
1 parent 39510a3
commit 01a02f6
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
@@ -157,8 +157,12 @@ jobs:
         run: |
           RUN_PRIVATE=$IS_NOT_FORK LUDWIG_TEST_SUITE_TIMEOUT_S=5400 pytest -v --timeout 300 --durations 100 -m "$MARKERS and not slow and not combinatorial and not horovod or benchmark and not llm" --junitxml pytest.xml tests/regression_tests
 
+      # Skip Horovod installation for torch nightly.
+      # https://github.com/ludwig-ai/ludwig/issues/3468
       - name: Install Horovod if necessary
-        if: matrix.test-markers == 'distributed'
+        if: |
+          matrix.test-markers == 'distributed' &&
+          "$PYTORCH" != "nightly"
         env:
           HOROVOD_WITH_PYTORCH: 1
           HOROVOD_WITHOUT_MPI: 1
@@ -174,8 +178,12 @@ jobs:
           horovodrun --check-build
         shell: bash
 
+      # Skip Horovod tests for torch nightly.
+      # https://github.com/ludwig-ai/ludwig/issues/3468
       - name: Horovod Tests
-        if: matrix.test-markers == 'distributed'
+        if: |
+          matrix.test-markers == 'distributed' &&
+          "$PYTORCH" != "nightly"
         run: |
           RUN_PRIVATE=$IS_NOT_FORK LUDWIG_TEST_SUITE_TIMEOUT_S=5400 pytest -v --timeout 300 --durations 100 -m "$MARKERS and horovod and not slow and not combinatorial and not llm" --junitxml pytest.xml tests/