def _get_config_for_image()

in sagemaker_image_builder/main.py [0:0]

• 7 lines of code
• 4 McCabe index (conditional complexity)

def _get_config_for_image(target_version_dir: str, image_generator_config, force_rebuild) -> dict:
    if not os.path.exists(target_version_dir + "/" + image_generator_config["env_out_filename"]) or force_rebuild:
        return image_generator_config

    config_for_image = copy.deepcopy(image_generator_config)
    # Use the existing env.out to create the conda environment. Pass that as env.in
    config_for_image["build_args"]["ENV_IN_FILENAME"] = image_generator_config["env_out_filename"]
    # Remove ARG_BASED_ENV_IN_FILENAME if it exists
    config_for_image["build_args"].pop("ARG_BASED_ENV_IN_FILENAME", None)
    return config_for_image