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)
def efs_fsx_setup(sagemaker_session):
fs_resources = set_up_efs_fsx(sagemaker_session)
try:
yield fs_resources
finally:
tear_down(sagemaker_session, fs_resources)