Function _attr_fields_dict_cache

axlearn/common/config.py:460–469

Cache the fields dict for type. Args: type_obj: Type to be cached. Returns: A dictionary of fields for the type.

(type_obj: type)

@cache
def _attr_fields_dict_cache(type_obj: type) -> dict[str, attr.Attribute]:
    """Cache the fields dict for type.

    Args:
        type_obj: Type to be cached.

    Returns:
        A dictionary of fields for the type.
    """
    return attr.fields_dict(type_obj)


@cache