Skip to content

Commit

Permalink
pre-commit applied
Browse files Browse the repository at this point in the history
  • Loading branch information
mdaniowi authored and mdaniowi committed Aug 12, 2024
1 parent ba5c41f commit 654bf15
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions tests/custom_op/test_attr.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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:]

0 comments on commit 654bf15

Please sign in to comment.