From 255c6304f73e594c71d13291962935ad04b244db Mon Sep 17 00:00:00 2001 From: reuvenp Date: Wed, 15 Jan 2025 12:00:19 +0200 Subject: [PATCH] fix imports in tutorials --- .../mct_features_notebooks/keras/example_keras_qat.ipynb | 1 + 1 file changed, 1 insertion(+) diff --git a/tutorials/notebooks/mct_features_notebooks/keras/example_keras_qat.ipynb b/tutorials/notebooks/mct_features_notebooks/keras/example_keras_qat.ipynb index 746e6b282..7d92e6be1 100644 --- a/tutorials/notebooks/mct_features_notebooks/keras/example_keras_qat.ipynb +++ b/tutorials/notebooks/mct_features_notebooks/keras/example_keras_qat.ipynb @@ -189,6 +189,7 @@ }, "outputs": [], "source": [ + "from mct_quantizers import QuantizationMethod\n", "from model_compression_toolkit.constants import FLOAT_BITWIDTH\n", "from model_compression_toolkit.target_platform_capabilities.constants import KERNEL_ATTR, BIAS_ATTR\n", "from model_compression_toolkit.target_platform_capabilities.schema.mct_current_schema import schema\n",