test: add retry for unity catalog tests

Signed-off-by: Omkar P <[email protected]>
delta-io · Feb 24, 2025 · 0f7282d · 0f7282d
1 parent 750f846
commit 0f7282d
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/.github/workflows/python_build.yml b/.github/workflows/python_build.yml
@@ -137,7 +137,12 @@ jobs:
         run: make setup-dat
 
       - name: Run tests
-        run: uv run --no-sync pytest -m '(unitycatalog_databricks and integration)' --doctest-modules
+        uses: nick-fields/retry@v3
+        with:
+          timeout_seconds: 10
+          max_attempts: 3
+          retry_on: error
+          command: uv run --no-sync pytest -m '(unitycatalog_databricks and integration)' --doctest-modules
 
   test-unitycatalog-oss:
     name: Python Build (Python 3.10 Unity Catalog Integration tests)
@@ -166,7 +171,12 @@ jobs:
         run: make setup-dat
 
       - name: Run tests
-        run: uv run --no-sync pytest -m '(unitycatalog_oss and integration)' --doctest-modules
+        uses: nick-fields/retry@v3
+        with:
+          timeout_seconds: 10
+          max_attempts: 3
+          retry_on: error
+          command: uv run --no-sync pytest -m '(unitycatalog_oss and integration)' --doctest-modules
 
   test-pyspark:
     name: PySpark Integration Tests