def get_sagemaker_mode()

in source/env_setup.py [0:0]

• 8 lines of code
• 2 McCabe index (conditional complexity)

def get_sagemaker_mode() -> str:
    stack_outputs_file = Path(CURRENT_FOLDER, 'stack_outputs.json')
    with open(stack_outputs_file) as f:
        outputs = json.load(f)
    sagemaker_mode = outputs['SagemakerMode']
    if sagemaker_mode not in set(['Studio', 'NotebookInstance']):
        raise ValueError('SagemakerMode should be Studio or NotebookInstance. Check stack_outputs.json.')
    return sagemaker_mode