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)