def _get_package_to_image_type_mapping()

in src/release_notes_generator.py [0:0]

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

def _get_package_to_image_type_mapping(image_type_package_metadata):
    # Creating a reverse dict based on the package names
    # { 'somepackage': {'cpu': '1.2.0', 'gpu': '1.2.2'}, ..}
    package_to_image_type_mapping = {}
    for image_type in image_type_package_metadata.keys():
        package_metadata = image_type_package_metadata[image_type]
        for package in package_metadata:
            if package not in package_to_image_type_mapping:
                package_to_image_type_mapping[package] = {image_type: package_metadata[package]}
            else:
                package_to_image_type_mapping[package][image_type] = package_metadata[package]
    return package_to_image_type_mapping