def _get_installed_packages()

in sagemaker_image_builder/release_notes_generator.py [0:0]

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

def _get_installed_packages(target_version_dir, image_config) -> dict[str, str]:
    env_in_file_name = image_config["build_args"]["ENV_IN_FILENAME"]
    env_out_file_name = image_config["env_out_filename"]
    env_in_file_path = target_version_dir + "/" + env_in_file_name
    if not os.path.exists(env_in_file_path):
        return {}
    required_packages_from_target = get_match_specs(target_version_dir + "/" + env_in_file_name).keys()
    target_match_spec_out = get_match_specs(target_version_dir + "/" + env_out_file_name)
    # We only care about the packages which are present in the target version env.in file
    return {
        k: str(v.get("version")).removeprefix("==")
        for k, v in target_match_spec_out.items()
        if k in required_packages_from_target
    }