From 569dc57da38899071382556f59abf44f55e5e4e1 Mon Sep 17 00:00:00 2001 From: Trevor Morris Date: Wed, 15 Apr 2020 18:02:57 +0000 Subject: [PATCH] Use tvm.transform.Sequential --- tests/python/relay/test_pass_partition_graph.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/python/relay/test_pass_partition_graph.py b/tests/python/relay/test_pass_partition_graph.py index 65d4c8f6f866..8827fbf1b8b0 100644 --- a/tests/python/relay/test_pass_partition_graph.py +++ b/tests/python/relay/test_pass_partition_graph.py @@ -1083,7 +1083,7 @@ def expected(): mod = tvm.IRModule() mod["main"] = create_graph() - seq = transform.Sequential([ + seq = tvm.transform.Sequential([ transform.AnnotateTarget(target), transform.MergeCompilerRegions(), transform.PartitionGraph(), @@ -1160,7 +1160,7 @@ def expected(): mod = tvm.IRModule() mod["main"] = create_graph() - seq = transform.Sequential([ + seq = tvm.transform.Sequential([ transform.AnnotateTarget(target), transform.MergeCompilerRegions(), transform.PartitionGraph(),