From dbe2918c18189ac103652710e4f4ceb328b50324 Mon Sep 17 00:00:00 2001 From: Maxim Zhiltsov Date: Tue, 28 Jan 2025 13:17:31 +0300 Subject: [PATCH] Make transform policy kw-only --- src/datumaro/components/dataset_storage.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/datumaro/components/dataset_storage.py b/src/datumaro/components/dataset_storage.py index 3228096b4c..ea6f89cfd9 100644 --- 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: