From 84510b3894f1674bf06294737ebd90cee00a6cb2 Mon Sep 17 00:00:00 2001 From: Lily Orth-Smith Date: Thu, 6 Jan 2022 14:14:52 -0800 Subject: [PATCH] tests --- tests/python/relay/test_json_compact.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) 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,