fix imports

Author: christinadionysio

src/main/python/systemds/scuro/__init__.py

@@ -116,7 +116,7 @@
     OverlappingSplitIndices,
 )
 from systemds.scuro.representations.elmo import ELMoRepresentation
-from systemds.scuro.representations.mlp_sampling import MLPSampling
+
 
 __all__ = [
     "BaseLoader",
@@ -200,5 +200,6 @@
     "SentenceBoundarySplit",
     "OverlappingSplit",
     "ELMoRepresentation",
-    "MLPSampling",
+    "SentenceBoundarySplitIndices",
+    "OverlappingSplitIndices",
 ]

src/main/python/systemds/scuro/drsearch/unimodal_optimizer.py

@@ -90,7 +90,6 @@ def _get_not_self_contained_reps(self, modality_type):
     def _get_context_operators(self, modality_type):
         return self.operator_registry.get_context_operators(modality_type)
 
-
     def store_results(self, file_name=None):
         if file_name is None:
             import time

src/main/python/systemds/scuro/representations/text_context_with_indices.py

@@ -75,8 +75,8 @@ def _extract_text(instance: Any) -> str:
     return text
 
 
-@register_context_operator(ModalityType.TEXT)
-class WordCountSplit(Context):
+# @register_context_operator(ModalityType.TEXT)
+class WordCountSplitIndices(Context):
     """
     Splits text after a fixed number of words.
 
@@ -135,7 +135,7 @@ def execute(self, modality):
 
 
 @register_context_operator(ModalityType.TEXT)
-class SentenceBoundarySplit(Context):
+class SentenceBoundarySplitIndices(Context):
     """
     Splits text at sentence boundaries while respecting maximum word count.
 
@@ -231,7 +231,7 @@ def execute(self, modality):
 
 
 @register_context_operator(ModalityType.TEXT)
-class OverlappingSplit(Context):
+class OverlappingSplitIndices(Context):
     """
     Splits text with overlapping chunks using a sliding window approach.