Merge branch 'master' into master

metrizable · web-flow · commit 644ac6d67ffc · 2021-01-19T10:44:21.000-08:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,11 @@
 # Changelog
 
+## v2.23.5 (2021-01-18)
+
+### Bug Fixes and Other Changes
+
+ * increase time allowed for trial components to index
+
 ## v2.23.4.post0 (2021-01-14)
 
 ### Documentation Changes
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-2.23.5.dev0
+2.23.6.dev0
diff --git a/tests/integ/sagemaker/lineage/test_artifact.py b/tests/integ/sagemaker/lineage/test_artifact.py
@@ -78,10 +78,12 @@ def test_list(artifact_objs, sagemaker_session):
 
 
 def test_downstream_trials(trial_associated_artifact, trial_obj, sagemaker_session):
-    # wait for TC to index
-    time.sleep(3)
-
-    trials = trial_associated_artifact.downstream_trials(sagemaker_session=sagemaker_session)
+    # allow trial components to index, 30 seconds max
+    for i in range(3):
+        time.sleep(10)
+        trials = trial_associated_artifact.downstream_trials(sagemaker_session=sagemaker_session)
+        if len(trials) > 0:
+            break
 
     assert len(trials) == 1
     assert trial_obj.trial_name in trials
