From 69140001204f47192432802ee9c873de9447f4a7 Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Tue, 2 Jul 2024 02:09:55 -0500 Subject: [PATCH] Fix formatting --- shark_turbine/transforms/general/add_metadata.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/shark_turbine/transforms/general/add_metadata.py b/shark_turbine/transforms/general/add_metadata.py index fae4f5ea..00d7c69c 100644 --- 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