Updated imports with the new custom layer package name (#1440)

Author: Idan-BenAmi

model_compression_toolkit/core/pytorch/reader/reader.py

@@ -23,7 +23,7 @@
 from model_compression_toolkit.core.common import Graph
 from model_compression_toolkit.core.pytorch.reader.graph_builders import edges_builder, nodes_builder
 from model_compression_toolkit.core.pytorch.utils import set_model
-from sony_custom_layers.pytorch import CustomLayer
+from edgemdt_cl.pytorch import CustomLayer
 
 
 def _trace_model(root: Union[torch.nn.Module, Callable[..., Any]]) -> GraphModule:

model_compression_toolkit/target_platform_capabilities/targetplatform2framework/attach2keras.py

@@ -20,7 +20,7 @@
 from model_compression_toolkit.target_platform_capabilities.targetplatform2framework.attach2fw import \
     AttachTpcToFramework
 
-from sony_custom_layers.keras.object_detection.ssd_post_process import SSDPostProcess
+from edgemdt_cl.keras.object_detection.ssd_post_process import SSDPostProcess
 
 if version.parse(tf.__version__) >= version.parse("2.13"):
     from keras.src.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, Dropout, \

model_compression_toolkit/target_platform_capabilities/targetplatform2framework/attach2pytorch.py

@@ -32,7 +32,7 @@
 from model_compression_toolkit.target_platform_capabilities.targetplatform2framework.attach2fw import \
     AttachTpcToFramework
 from model_compression_toolkit.target_platform_capabilities.targetplatform2framework.attribute_filter import Eq
-from sony_custom_layers.pytorch import MulticlassNMS, MulticlassNMSWithIndices
+from edgemdt_cl.pytorch import MulticlassNMS, MulticlassNMSWithIndices
 
 
 class AttachTpcToPytorch(AttachTpcToFramework):

requirements.txt

@@ -11,5 +11,5 @@ scipy
 protobuf
 mct-quantizers-nightly
 pydantic>=2.0
-sony-custom-layers-dev==0.4.0.dev6
+edge-mdt-cl-dev
 

tests/keras_tests/custom_layers_tests/test_sony_ssd_postprocess_layer.py

@@ -18,7 +18,7 @@
 import tensorflow as tf
 
 import model_compression_toolkit as mct
-from sony_custom_layers.keras.object_detection.ssd_post_process import SSDPostProcess
+from edgemdt_cl.keras.object_detection.ssd_post_process import SSDPostProcess
 from mct_quantizers.keras.metadata import MetadataLayer
 from tests.common_tests.helpers.tpcs_for_tests.v4.tpc import get_tpc