fix keras input layer import in tests

sony · Jan 1, 2025 · 42be1cb · 42be1cb
1 parent d0e6fd6
commit 42be1cb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tests/keras_tests/function_tests/test_sensitivity_metric_interest_points.py b/tests/keras_tests/function_tests/test_sensitivity_metric_interest_points.py
@@ -17,7 +17,7 @@
 import numpy as np
 from keras.applications.densenet import DenseNet121
 from keras.applications.mobilenet_v2 import MobileNetV2
-from keras_core.src.layers import InputLayer
+from keras.src.engine.input_layer import InputLayer
 
 from packaging import version
 

diff --git a/tests/keras_tests/function_tests/test_set_layer_to_bitwidth.py b/tests/keras_tests/function_tests/test_set_layer_to_bitwidth.py
@@ -18,7 +18,7 @@
 
 from keras import Input
 from keras.layers import Conv2D
-from keras_core.src.layers import InputLayer
+from keras.src.engine.input_layer import InputLayer
 
 from mct_quantizers import KerasActivationQuantizationHolder
 from model_compression_toolkit.core import QuantizationConfig

diff --git a/tests/keras_tests/function_tests/test_weights_activation_split_substitution.py b/tests/keras_tests/function_tests/test_weights_activation_split_substitution.py
@@ -19,7 +19,7 @@
 
 from keras.layers import Conv2D, Conv2DTranspose, DepthwiseConv2D, Dense, BatchNormalization, ReLU, Input
 import numpy as np
-from keras_core.src.layers import InputLayer
+from keras.src.engine.input_layer import InputLayer
 
 from model_compression_toolkit.core import QuantizationConfig
 from model_compression_toolkit.core.common.graph.virtual_activation_weights_node import VirtualSplitActivationNode, \