def register_dataclass()

in neuron-explainer/neuron_explainer/fast_dataclasses/fast_dataclasses.py [0:0]

• 6 lines of code
• 3 McCabe index (conditional complexity)

def register_dataclass(cls):  # type: ignore
    assert is_dataclass(cls), "Only dataclasses can be registered."
    dataclasses_by_name[cls.__name__] = cls
    name_set = frozenset(f.name for f in fields(cls) if f.name != "dataclass_name")
    dataclasses_by_fieldnames[name_set] = cls
    return cls