fix: add time for search to get updated

Author: ahsan-z-khan

tests/integ/test_experiments_analytics.py

@@ -36,6 +36,9 @@ def experiment(sagemaker_session):
                 TrialComponentName=trial_component_name, TrialName=trial_name
             )
 
+        time.sleep(15)  # wait for search to get updated
+
+        # allow search time thrice
         for _ in range(3):
             analytics = ExperimentAnalytics(
                 experiment_name=experiment_name, sagemaker_session=sagemaker_session
@@ -44,7 +47,7 @@ def experiment(sagemaker_session):
             if len(analytics.dataframe().columns) > 0:
                 break
 
-            time.sleep(15)  # wait for search to get updated
+            time.sleep(15)
 
         yield experiment_name
     finally:
@@ -85,6 +88,9 @@ def experiment_with_artifacts(sagemaker_session):
                 TrialComponentName=trial_component_name, TrialName=trial_name
             )
 
+        time.sleep(15)  # wait for search to get updated
+
+        # allow search time thrice
         for _ in range(3):
             analytics = ExperimentAnalytics(
                 experiment_name=experiment_name, sagemaker_session=sagemaker_session
@@ -93,7 +99,7 @@ def experiment_with_artifacts(sagemaker_session):
             if len(analytics.dataframe().columns) > 0:
                 break
 
-            time.sleep(15)  # wait for search to get updated
+            time.sleep(15)
 
         yield experiment_name
     finally: