[COLAB] No module named 'tensorflow_model_optimization'

[rjtp5670]

Prior to filing: check that this should be a bug instead of a feature request. Everything supported, including the compatible versions of TensorFlow, is listed in the overview page of each technique. For example, the overview page of quantization-aware training is here. An issue for anything not supported should be a feature request.

Describe the bug A clear and concise description of what the bug is.

🗯️ ModuleNotFoundError: No module named 'tensorflow_model_optimization'

System information

TensorFlow version (installed from source or binary): 2.13.0

TensorFlow Model Optimization version (installed from source or binary): Can not be found.

Python version: 3.10.12 (main, Jun 11 2023, 05:26:28) [GCC 11.4.0]

• If this behavior was never correct, please file a report for a regular bug, not a regression.
• (Optional) Commit/PR that introduced the regression:

Describe the expected behavior

• Importing tfmot Library

Describe the current behavior

• Failed to import the library although tensorflow_model_optimization installed.

Code to reproduce the issue (Running a code in Colab)

!pip install -q tensorflow-model-optimization import tensorflow_model_optimization as tfmot

Screenshots

Additional context Add any other context about the problem here.

(Thank you so much for your effort and reading my report. )

I've gone through prior issue reports but nothing was found to resolve that issue in Colab.

[rjtp5670]

Additional Report

• I restarted my runtime from CPU to A100 and it no longer throw the importing error. And I came back to CPU runtime, but I can not reproduce issue. Maybe it has something to do with my disordering code cells. Not it works fine.