From 654bf1526075f4335b733cefc64e660e42db4b15 Mon Sep 17 00:00:00 2001 From: mdaniowi Date: Mon, 12 Aug 2024 08:31:07 +0000 Subject: [PATCH] pre-commit applied --- tests/custom_op/test_attr.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/tests/custom_op/test_attr.py b/tests/custom_op/test_attr.py index 6e2527ac..cde5a321 100644 --- a/tests/custom_op/test_attr.py +++ b/tests/custom_op/test_attr.py @@ -37,10 +37,7 @@ class AttrTestOp(CustomOp): def get_nodeattr_types(self): - my_attrs = { - "tensor_attr": ("t", True, np.asarray([])), - "strings_attr": ("strings", True, [""]) - } + my_attrs = {"tensor_attr": ("t", True, np.asarray([])), "strings_attr": ("strings", True, [""])} return my_attrs def make_shape_compatible_op(self, model): @@ -105,4 +102,3 @@ def test_attr(): strings_attr_prod[0] = "test" inst.set_nodeattr("strings_attr", strings_attr_prod) assert inst.get_nodeattr("strings_attr") == ["test"] + strings_attr[1:] -