def load_sagemaker_config()

in src/helpers.py [0:0]

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

def load_sagemaker_config(args: dict) -> dict:
    """
    Load SageMaker config from `/opt/ml/input/config/hyperparameters.json`
    """
    file_path = "/opt/ml/input/config/hyperparameters.json"
    if os.path.isfile(file_path):
        with open(file_path, "r") as f:
            new_args = json.load(f)
            for k, v in new_args.items():
                if k not in args:
                    continue
                if isinstance(args[k], int):
                    v = int(v)
                if isinstance(args[k], float):
                    v = float(v)
                args[k] = v
    return args