diff --git a/external/mmdetection/detection_tasks/extension/datasets/mmdataset.py b/external/mmdetection/detection_tasks/extension/datasets/mmdataset.py
index ccc280c8440..8a73da659ba 100644
--- a/external/mmdetection/detection_tasks/extension/datasets/mmdataset.py
+++ b/external/mmdetection/detection_tasks/extension/datasets/mmdataset.py
@@ -139,12 +139,13 @@ def __getitem__(self, index):
 
     @check_input_parameters_type({"ote_dataset": DatasetParamTypeCheck})
     def __init__(
-            self,
-            ote_dataset: DatasetEntity,
-            labels: List[LabelEntity],
-            pipeline: Sequence[dict],
-            domain: Domain,
-            test_mode: bool = False,
+        self,
+        ote_dataset: DatasetEntity,
+        labels: List[LabelEntity],
+        pipeline: Sequence[dict],
+        domain: Domain,
+        test_mode: bool = False,
+        **kwargs,
     ):
         self.ote_dataset = ote_dataset
         self.labels = labels
diff --git a/external/mmsegmentation/segmentation_tasks/extension/datasets/mmdataset.py b/external/mmsegmentation/segmentation_tasks/extension/datasets/mmdataset.py
index aa5eb645bf9..5592ce260e2 100644
--- a/external/mmsegmentation/segmentation_tasks/extension/datasets/mmdataset.py
+++ b/external/mmsegmentation/segmentation_tasks/extension/datasets/mmdataset.py
@@ -82,7 +82,12 @@ class _DataInfoProxy:
         forwards data access operations to ote_dataset and converts the dataset items to the view
         convenient for mmsegmentation.
         """
-        def __init__(self, ote_dataset, labels=None):
+        def __init__(
+            self,
+            ote_dataset,
+            labels=None,
+            **kwargs,
+        ):
             self.ote_dataset = ote_dataset
             self.labels = labels
             self.label_idx = {label.id: i for i, label in enumerate(labels)}