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

@@ -13,6 +13,7 @@
 from __future__ import absolute_import
 
 import os
+import sys
 
 import pytest
 import numpy as np
@@ -54,6 +55,7 @@
     seconds_to_sleep=6,
 ):
     try:
+        print(sys.modules)
         from airflow import utils
         from airflow import DAG
         from airflow.providers.amazon.aws.operators.sagemaker import SageMakerTrainingOperator
@@ -64,6 +66,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")