Skip to content

Commit 6e977c2

Browse files
committed
CI fix.
1 parent 5cfeda0 commit 6e977c2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/python/frontend/tflite/test_forward.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
from tensorflow.python.ops import variables
1717
from tensorflow.contrib.lite.python import interpreter as interpreter_wrapper
1818

19-
import nnvm.testing.tf
19+
import tvm.relay.testing.tf as tf_testing
2020

2121
#######################################################################
2222
# Generic run functions for TVM & TFLite
@@ -344,7 +344,7 @@ def test_forward_mobilenet():
344344
'''test mobilenet v1 tflite model'''
345345
# MobilenetV1
346346
temp = util.tempdir()
347-
tflite_model_file = nnvm.testing.tf.get_workload_official(
347+
tflite_model_file = tf_testing.get_workload_official(
348348
"http://download.tensorflow.org/models/mobilenet_v1_2018_08_02/mobilenet_v1_1.0_224.tgz",
349349
"mobilenet_v1_1.0_224.tflite", temp)
350350
tflite_model_buf = open(tflite_model_file, "rb").read()

0 commit comments

Comments
 (0)