bff50cfc37380140d53debfb140e262a38668db0,python/ray/autoscaler/_private/kubernetes/config.py,,_get_resource,#Any#Any#Any#,102

Before Change

def _get_resource(container_resources, resource_name, field_name):
    if (field_name in container_resources
            and resource_name in container_resources[field_name]):
        return _parse_resource(container_resources[field_name][resource_name])
    else:
        return float("inf")


def _parse_resource(resource):
    resource_str = str(resource)
    if resource_str[-1] == "m":
        return math.ceil(int(resource_str[:-1]) / 1000)

After Change

    resources = container_resources[field_name]
    // Look for keys containing the resource_name. For example,
    // the key "nvidia.com/gpu" contains the key "gpu".
    matching_keys = [key for key in resources if resource_name in key.lower()]
    if len(matching_keys) == 0:
        return float("inf")
    if len(matching_keys) > 1:
        // Should have only one match -- mostly relevant for gpu.
        raise ValueError(f"Multiple {resource_name} types not supported.")
    // E.g. "nvidia.com/gpu" or "cpu".
    resource_key = matching_keys.pop()
    resource_quantity = resources[resource_key]
    return _parse_resource(resource_quantity)

Instances