Cleanup formatting in ir_data_fields
diff --git a/compiler/util/ir_data_fields.py b/compiler/util/ir_data_fields.py
index 86b670c..c74bd8f 100644
--- a/compiler/util/ir_data_fields.py
+++ b/compiler/util/ir_data_fields.py
@@ -108,7 +108,6 @@
class TemporaryCopyValuesList(NamedTuple):
"""Class used to temporarily hold a CopyValuesList while copying and
-
constructing an IR dataclass.
"""
@@ -197,6 +196,7 @@
if isinstance(type, v.__class__) and _is_ir_dataclass(v)
)
+
class IrDataclassSpecs:
"""Maintains a cache of all IR dataclass specs."""
@@ -206,8 +206,8 @@
def get_mod_specs(cls, mod):
"""Gets the IR dataclass specs for the given module."""
return {
- ir_class: FilteredIrFieldSpecs(_field_specs(ir_class))
- for ir_class in all_ir_classes(mod)
+ ir_class: FilteredIrFieldSpecs(_field_specs(ir_class))
+ for ir_class in all_ir_classes(mod)
}
@classmethod
@@ -228,9 +228,7 @@
"""
for data_class in all_ir_classes(mod):
if data_class is not cls:
- data_class.field_specs = IrDataclassSpecs.get_specs(
- data_class
- )
+ data_class.field_specs = IrDataclassSpecs.get_specs(data_class)
def _field_specs(cls: type[IrDataT]) -> Mapping[str, FieldSpec]:
