Retry ValueError for airflow tests

This change adds a retry to import utils from airflow in case it trys to
load util from local module instead of airflow dependency

Author: navinsoni

tests/integ/test_airflow_config.py

@@ -64,6 +64,11 @@
         break
     except ParsingError:
         pass
+    except ValueError as ve:
+        if "Unable to configure formatter" in str(ve):
+            print(f"Received: {ve}")
+        else:
+            raise ve
 
 PYTORCH_MNIST_DIR = os.path.join(DATA_DIR, "pytorch_mnist")
 PYTORCH_MNIST_SCRIPT = os.path.join(PYTORCH_MNIST_DIR, "mnist.py")