Change import to make it more specific

gussmith23 · Oct 17, 2019 · 79bf21f · 79bf21f
1 parent c45684d
commit 79bf21f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/unittest/test_custom_datatypes_change_dtype.py b/tests/python/unittest/test_custom_datatypes_change_dtype.py
@@ -19,7 +19,7 @@
 import tvm
 import numpy as np
 from tvm import relay
-from tvm.relay.testing.inception_v3 import get_workload
+from tvm.relay.testing.inception_v3 import get_workload as get_inception
 
 tgt = "llvm"
 
@@ -94,7 +94,7 @@ def test_change_dtype_simple():
 
 def test_change_dtype_inception_v3():
 
-    expr, params = get_workload()
+    expr, params = get_inception()
 
     ex = relay.create_executor("graph")