feat: added interface for the processing strategies

Modalities · Jan 5, 2025 · 8f41a74 · 8f41a74
1 parent 3d79cc4
commit 8f41a74
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/modalities/dataloader/preprocessing/queued_processing/__init__.py b/src/modalities/dataloader/preprocessing/queued_processing/__init__.py
diff --git a/src/modalities/dataloader/preprocessing/queued_processing/processing_strategy_if.py b/src/modalities/dataloader/preprocessing/queued_processing/processing_strategy_if.py
@@ -0,0 +1,16 @@
+from abc import ABC
+from typing import Any
+
+
+class ProcessingStrategyIF(ABC):
+    def process(self, item: Any) -> dict[str, Any] | None:
+        raise NotImplementedError
+
+    def __enter__(self):
+        raise NotImplementedError
+
+    def finalize(self):
+        raise NotImplementedError
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        raise