code: def efs_fsx_setup(sagemaker_session, ec2_instance_type):
      fs_resources = None
      try:
          fs_resources = set_up_efs_fsx(
              sagemaker_session, ec2_instance_type
          )
          yield fs_resources
      finally:
          if fs_resources:
              tear_down(sagemaker_session, fs_resources)
  
file_path: sagemaker-python-sdk/tests/integ/test_kmeans_efs_fsx.py
imports: from __future__ import absolute_import
  import pytest
  from sagemaker import KMeans
  from sagemaker.amazon.amazon_estimator import FileSystemRecordSet
  from sagemaker.parameter import IntegerParameter, CategoricalParameter
  from sagemaker.tuner import HyperparameterTuner
  from sagemaker.utils import unique_name_from_base
  import tests
  from tests.integ import TRAINING_DEFAULT_TIMEOUT_MINUTES, TUNING_DEFAULT_TIMEOUT_MINUTES
  from tests.integ.file_system_input_utils import set_up_efs_fsx, tear_down
  from tests.integ.s3_utils import assert_s3_files_exist
  from tests.integ.timeout import timeout
label_desc: Use image node_id 1 for calling a global function with example usage: efs_fsx_setup(sagemaker_session, ec2_instance_type) without return types