Make transform policy kw-only

cvat-ai · Jan 28, 2025 · dbe2918 · dbe2918
1 parent 64d0020
commit dbe2918
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/datumaro/components/dataset_storage.py b/src/datumaro/components/dataset_storage.py
@@ -87,6 +87,7 @@ def __init__(
         self,
         source: IDataset,
         transforms: List[PostponedTransform],
+        *,
         raise_on_malformed_transform: bool = True,
     ):
         super().__init__(source)
@@ -138,6 +139,7 @@ def __init__(
         categories: Optional[CategoriesInfo] = None,
         media_type: Optional[Type[MediaElement]] = None,
         ann_types: Optional[Set[AnnotationType]] = None,
+        *,
         raise_on_malformed_transform: bool = True,
     ):
         self._raise_on_malformed_transform = raise_on_malformed_transform
@@ -272,7 +274,9 @@ def _add_ann_types(item: DatasetItem):
         old_ids = None
         if self._transforms:
             transform = _StackedTransform(
-                source, self._transforms, self._raise_on_malformed_transform
+                source, 
+                self._transforms, 
+                raise_on_malformed_transform=self._raise_on_malformed_transform
             )
             if transform.is_local:
                 # An optimized way to find modified items:
@@ -675,6 +679,7 @@ def __init__(
         categories: Optional[CategoriesInfo] = None,
         media_type: Optional[Type[MediaElement]] = None,
         ann_types: Optional[Set[AnnotationType]] = None,
+        *,
         raise_on_malformed_transform: bool = True,
     ):
         if not source.is_stream: