fix error handling

Author: laurenyu

tests/integ/test_kmeans_efs_fsx.py

@@ -38,8 +38,8 @@
 
 @pytest.fixture(scope="module")
 def efs_fsx_setup(sagemaker_session):
-    fs_resources = set_up_efs_fsx(sagemaker_session)
     try:
+        fs_resources = set_up_efs_fsx(sagemaker_session)
         yield fs_resources
     finally:
         tear_down(sagemaker_session, fs_resources)

tests/integ/test_tf_efs_fsx.py

@@ -41,8 +41,8 @@
 
 @pytest.fixture(scope="module")
 def efs_fsx_setup(sagemaker_session):
-    fs_resources = set_up_efs_fsx(sagemaker_session)
     try:
+        fs_resources = set_up_efs_fsx(sagemaker_session)
         yield fs_resources
     finally:
         tear_down(sagemaker_session, fs_resources)