Move the minio artifact download under try block (#3889)

If minio artifact download fails the workteam is not being deleted.
kubeflow · Jun 2, 2020 · a7be049 · a7be049
1 parent 58f1d13
commit a7be049
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/components/aws/sagemaker/tests/integration_tests/component_tests/test_workteam_component.py b/components/aws/sagemaker/tests/integration_tests/component_tests/test_workteam_component.py
@@ -57,11 +57,12 @@ def test_workteamjob(
     )
 
     outputs = {"sagemaker-private-workforce": ["workteam_arn"]}
-    output_files = minio_utils.artifact_download_iterator(
-        workflow_json, outputs, download_dir
-    )
 
-    try:
+    try: 
+        output_files = minio_utils.artifact_download_iterator(
+            workflow_json, outputs, download_dir
+        )
+
         response = sagemaker_utils.describe_workteam(sagemaker_client, workteam_name)
 
         # Verify WorkTeam was created in SageMaker