From 79bf21ff19b793ea71275ea7e0428b36d6da28be Mon Sep 17 00:00:00 2001 From: Gus Smith Date: Thu, 16 May 2019 17:24:36 -0700 Subject: [PATCH] Change import to make it more specific --- tests/python/unittest/test_custom_datatypes_change_dtype.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/python/unittest/test_custom_datatypes_change_dtype.py b/tests/python/unittest/test_custom_datatypes_change_dtype.py index 56fdbc77dee9..a13f912141da 100644 --- 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")