Skip to content

Commit 654bf15

Browse files
mdaniowimdaniowi
authored andcommitted
pre-commit applied
1 parent ba5c41f commit 654bf15

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

tests/custom_op/test_attr.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -37,10 +37,7 @@
3737

3838
class AttrTestOp(CustomOp):
3939
def get_nodeattr_types(self):
40-
my_attrs = {
41-
"tensor_attr": ("t", True, np.asarray([])),
42-
"strings_attr": ("strings", True, [""])
43-
}
40+
my_attrs = {"tensor_attr": ("t", True, np.asarray([])), "strings_attr": ("strings", True, [""])}
4441
return my_attrs
4542

4643
def make_shape_compatible_op(self, model):
@@ -105,4 +102,3 @@ def test_attr():
105102
strings_attr_prod[0] = "test"
106103
inst.set_nodeattr("strings_attr", strings_attr_prod)
107104
assert inst.get_nodeattr("strings_attr") == ["test"] + strings_attr[1:]
108-

0 commit comments

Comments
 (0)