Fix formatting

iree-org · Jul 2, 2024 · 6914000 · 6914000
1 parent f196208
commit 6914000
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/shark_turbine/transforms/general/add_metadata.py b/shark_turbine/transforms/general/add_metadata.py
@@ -51,7 +51,10 @@ def parse_metadata_dict(metadata_dict: dict) -> DictAttr:
         metadata_dict_attr = parse_metadata_dict(self.inp_metadata)
         for func_op in self.funcs:
             for attr in func_op.op.attributes:
-                if isinstance(attr.attr, StringAttr) and attr.attr.value == self.func_name:
+                if (
+                    isinstance(attr.attr, StringAttr)
+                    and attr.attr.value == self.func_name
+                ):
                     func_op.op.attributes["iree.reflection"] = metadata_dict_attr
         return self.mlir_module