diff --git a/tests/python/relay/test_json_compact.py b/tests/python/relay/test_json_compact.py index de7ab42326fb..dc035adda2c2 100644 --- a/tests/python/relay/test_json_compact.py +++ b/tests/python/relay/test_json_compact.py @@ -49,7 +49,13 @@ def test_var(): {"type_key": ""}, { "type_key": "relay.Var", - "attrs": {"_checked_type_": "0", "span": "0", "type_annotation": "0", "vid": "2", "virtual_device_": "0"}, + "attrs": { + "_checked_type_": "0", + "span": "0", + "type_annotation": "0", + "vid": "2", + "virtual_device_": "0", + }, }, {"type_key": "relay.Id", "attrs": {"name_hint": "a3"}}, {"type_key": "relay.TensorType", "attrs": {"dtype": "float32", "shape": "4", "span": "0"}}, @@ -130,7 +136,10 @@ def test_global_var(): assert isinstance(tvar, tvm.ir.GlobalVar) nodes = [ {"type_key": ""}, - {"type_key": "GlobalVar", "attrs": {"_checked_type_": "0", "name_hint": "x", "span": "0", "virtual_device_": "0"}}, + { + "type_key": "GlobalVar", + "attrs": {"_checked_type_": "0", "name_hint": "x", "span": "0", "virtual_device_": "0"}, + }, ] data = { "root": 1,