From 0f7282d733cda32e6b274c8353dad81518348f47 Mon Sep 17 00:00:00 2001
From: Omkar P <45419097+omkar-foss@users.noreply.github.com>
Date: Mon, 24 Feb 2025 18:32:52 +0530
Subject: [PATCH] test: add retry for unity catalog tests

Signed-off-by: Omkar P <45419097+omkar-foss@users.noreply.github.com>
---
 .github/workflows/python_build.yml | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/python_build.yml b/.github/workflows/python_build.yml
index 23ff65073d..5ab911f55b 100644
--- 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