restore parts of text printer testing

apache · Apr 11, 2019 · 7e558e6 · 7e558e6
1 parent 2dcdd39
commit 7e558e6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/relay/test_ir_text_printer.py b/tests/python/relay/test_ir_text_printer.py
@@ -50,8 +50,8 @@ def test_env():
     text = env.astext()
     assert "def @myf" in text
     assert "def @myf" in str(env)
-    assert "add(%0, %0)" in text
-    assert "add(%0, %0)" in str(env)
+    assert "add(%0, %0) /* ty=float32 */" in text
+    assert "add(%0, %0) /* ty=float32 */" in str(env)
     show(env.astext(annotate=lambda x: str(x.checked_type.dtype)))
     show(text)