* CI fixes

apache · Mar 6, 2021 · bd6e0b9 · bd6e0b9
1 parent 8ba2290
commit bd6e0b9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 12 deletions.
diff --git a/tests/python/frontend/tensorflow/test_forward_v2_x.py b/tests/python/frontend/tensorflow/test_forward_v2_x.py
@@ -762,7 +762,7 @@ def test_forward_biasadd():
     _test_biasadd([4, 3, 3, 124], "NHWC")
 
 
-'''
+"""
 def _test_forward_where(input_shape):
     with tf.Graph().as_default():
         dtype = tf.float32
@@ -780,7 +780,7 @@ def test_forward_argwhere():
     _test_forward_where((5, 5, 5))
     _test_forward_where((5, 5, 5, 5))
     _test_forward_where((5, 5, 5, 5, 5))
-'''
+"""
 
 
 #######################################################################
@@ -990,7 +990,7 @@ def test_forward_squeeze():
     _test_squeeze(np.arange(6).reshape((1, 2, 1, 3, 1)), [-3, -5, -1])
 
 
-'''
+"""
 # TODO: The following operators are not implemented: {'TensorListGetItem', 'TensorListSetItem', 'TensorListReserve'}
 def test_tensor_array_constructor():
     def run(dtype_str):
@@ -1154,7 +1154,7 @@ def array_func():
         run(dtype, (5, 5, 5, 5))
         run(dtype, (5, 5, 5, 5, 5))
         run(dtype, (5, 5, 5, 5, 5, 5))
-'''
+"""
 
 
 #######################################################################
@@ -1700,7 +1700,7 @@ def test_forward_gather_nd():
     compare_tf_with_tvm_v2([input_data], concrete_func)
 
 
-'''
+"""
 #######################################################################
 # BiasAdd
 # -------
@@ -1718,7 +1718,7 @@ def check_bias_add(lh_shpae, rh_shape, dtype):
 
     check_bias_add((10, 8, 16, 32), (32,), dtype="int32")
     check_bias_add((10, 20), (20,), dtype="float32")
-'''
+"""
 
 
 #######################################################################
@@ -1902,7 +1902,7 @@ def _test_resize_nearest_neighbor_dynamic_shape(in_shape, scale):
     compare_tf_with_tvm_v2([input_data], concrete_func, opt_level=1)
 
 
-'''
+"""
 def test_forward_resize():
     """ Resize Bilinear, Nearest_Neighbor """
     # TF default layout is NHWC
@@ -1912,7 +1912,7 @@ def test_forward_resize():
     _test_resize_bilinear_from_tensor((6, 50, 50, 3), True)
     _test_resize_nearest_neighbor((6, 32, 32, 3), [20, 20])
     _test_resize_nearest_neighbor_dynamic_shape((1, 16, 16, 3), scale=[2, 2])
-'''
+"""
 
 
 #######################################################################
@@ -3265,7 +3265,7 @@ def test_forward_one_hot():
 # ----
 
 
-'''
+"""
 def _test_forward_add_n(inputs):
     root = tracking.AutoTrackable()
     root.f = def_function.function(lambda x: tf.add_n(x))
@@ -3292,7 +3292,7 @@ def test_forward_add_n():
     # _test_forward_add_n(in3)
     # _test_forward_add_n(in4)
     # _test_forward_add_n(in5)
-'''
+"""
 
 
 #######################################################################
@@ -3395,5 +3395,5 @@ def test_forward_add_n():
     # test_forward_lstm() - TODO
 
     # End to End
-    # test_forward_image_classification_models() # TODO: Need docker update.
+    test_forward_image_classification_models()
     # test_forward_ptb() - TODO
diff --git a/tests/scripts/task_ci_setup.sh b/tests/scripts/task_ci_setup.sh
@@ -28,10 +28,12 @@ set -o pipefail
 #
 # command: python3 -m pip install --user <package>==<version>
 #
-echo "Addtiional setup in" ${CI_IMAGE_NAME}
+echo "Additional setup in" ${CI_IMAGE_NAME}
 
 python3 -m pip install --user tlcpack-sphinx-addon==0.1.4 synr==0.2.1
 
+python3 -m pip install --user tensorflow-hub
+
 # Rebuild standalone_crt in build/ tree. This file is not currently archived by pack_lib() in
 # Jenkinsfile. We expect config.cmake to be present from pack_lib().
 # TODO(areusch): Make pack_lib() pack all the data dependencies of TVM.