* Docker update for tensorflow-hub

apache · Mar 1, 2021 · 48eba8c · 48eba8c
1 parent 413a1d0
commit 48eba8c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/docker/install/ubuntu_install_tensorflow.sh b/docker/install/ubuntu_install_tensorflow.sh
@@ -24,3 +24,6 @@ set -o pipefail
 # tensorflow:
 # https://github.com/tensorflow/tensorflow/issues/44467
 pip3 install tensorflow==2.3.1 keras==2.4.3 "h5py<3.0"
+
+#tensorflow-hub
+pip3 install tensorflow-hub
diff --git a/tests/python/frontend/tensorflow/test_forward_v2_x.py b/tests/python/frontend/tensorflow/test_forward_v2_x.py
@@ -41,8 +41,6 @@
 from tensorflow.python.grappler import tf_optimizer
 from tensorflow.python.training.saver import export_meta_graph
 
-import tensorflow_hub as hub
-
 from tensorflow.python.ops import nn_ops
 from tensorflow.python.ops import nn
 from tensorflow.python.ops import array_ops
@@ -2248,6 +2246,7 @@ def test_forward_where():
 def test_forward_image_classification_models():
     """Test E2E models from Tensorflow Hub"""
 
+    import tensorflow_hub as hub
     models = [
         # Inception V1
         {
@@ -3385,5 +3384,5 @@ def test_forward_add_n():
     # test_forward_lstm() - TODO
 
     # End to End
-    test_forward_image_classification_models()
+    # test_forward_image_classification_models() # TODO: Need docker update.
     # test_forward_ptb() - TODO