Remove tensorflow-model-optimization from documentation and scripts (#…

…739) Co-authored-by: Ofir Gordon <[email protected]>
sony · Jul 13, 2023 · 2bb2a86 · 2bb2a86
1 parent 0b082e3
commit 2bb2a86
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 7 deletions.
diff --git a/.github/workflows/run_tests_suite_pip_keras.yml b/.github/workflows/run_tests_suite_pip_keras.yml
@@ -26,7 +26,7 @@ jobs:
         run: |
           git checkout tags/${{  inputs.mct_version  }}
           pip install -r requirements.txt
-          pip install tensorflow tensorflow-model-optimization
+          pip install tensorflow
           pip install torch torchvision
       - name: Build WHL file
         run: |

diff --git a/INSTALLATION.md b/INSTALLATION.md
@@ -34,9 +34,8 @@ pip install -r requirements.txt
 
 Before running MCT, make sure to install one of the supported frameworks: TensorFlow or PyTorch.
 
-If you intend to use MCT with TensorFlow, install the following packages: 
-[tensorflow](https://www.tensorflow.org/install), 
-[tensorflow-model-optimization](https://www.tensorflow.org/model_optimization/guide/install)
+If you intend to use MCT with TensorFlow, install the following package: 
+[tensorflow](https://www.tensorflow.org/install)
 
 If you plan to use MCT with PyTorch, install the following package: 
 [torch](https://pytorch.org/)

diff --git a/...ssion_toolkit/target_platform_capabilities/tpc_models/get_target_platform_capabilities.py b/...ssion_toolkit/target_platform_capabilities/tpc_models/get_target_platform_capabilities.py
@@ -51,7 +51,7 @@ def get_target_platform_capabilities(fw_name: str,
     fw_tpc = tpc_dict.get(target_platform_name)
     assert fw_name in fw_tpc, f'Framework {fw_name} is not supported in {target_platform_name}. Please make sure the relevant ' \
                               f'packages are installed when using MCT for optimizing a {fw_name} model. ' \
-                              f'For Tensorflow, please install tensorflow and tensorflow-model-optimization. ' \
+                              f'For Tensorflow, please install tensorflow. ' \
                               f'For PyTorch, please install torch.'
     tpc_versions = fw_tpc.get(fw_name)
     if target_platform_version is None:

diff --git a/tutorials/notebooks/example_keras_imagenet.ipynb b/tutorials/notebooks/example_keras_imagenet.ipynb
@@ -97,7 +97,6 @@
       "outputs": [],
       "source": [
         "!pip install -q tensorflow\n",
-        "!pip install -q tensorflow-model-optimization\n",
         "!pip install -q model-compression-toolkit"
       ]
     },

diff --git a/tutorials/notebooks/example_keras_qat.ipynb b/tutorials/notebooks/example_keras_qat.ipynb
@@ -37,7 +37,6 @@
    "outputs": [],
    "source": [
     "! pip install -q tensorflow==2.10.1\n",
-	"! pip install -q tensorflow-model-optimization\n",
     "! pip install -q model-compression-toolkit "
    ]
   },