def env_setup()

in sagemaker_notebook_instance/env_setup.py [0:0]

• 21 lines of code
• 6 McCabe index (conditional complexity)

def env_setup() -> None:
    args = parse_args()
    logging_setup(args.log_level)
    sagemaker_mode = get_sagemaker_mode()
    if sagemaker_mode == 'Studio':
        hostname = get_hostname()
        logging.debug(f'hostname: {hostname}')
        executable = get_executable()
        logging.debug(f'executable: {executable}')
        if args.force or not in_logbook(hostname, executable):
            env_setup_studio()
            logging.info('Successfully setup environment.')
            add_to_logbook(hostname, executable)
        else:
             logging.info('Skipping. Already setup environment.')
    if sagemaker_mode == 'NotebookInstance':
        if args.force:
            env_setup_notebook_instance()
            logging.info('Successfully setup environment.')
        else:
            logging.info('Skipping. Already setup environment.')