sinapsis-data-readers 0.1.16__tar.gz → 0.1.17__tar.gz

{sinapsis_data_readers-0.1.16/src/sinapsis_data_readers.egg-info → sinapsis_data_readers-0.1.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-data-readers
-Version: 0.1.16
+Version: 0.1.17
 Summary: Templates to read data in different formats
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech

{sinapsis_data_readers-0.1.16 → sinapsis_data_readers-0.1.17}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sinapsis-data-readers"
-version = "0.1.16"
+version = "0.1.17"
 description = "Templates to read data in different formats"
 authors = [{ name = "SinapsisAI", email = "dev@sinapsis.tech" }]
 

{sinapsis_data_readers-0.1.16 → sinapsis_data_readers-0.1.17}/src/sinapsis_data_readers/templates/datasets_readers/sklearn_datasets.py

@@ -135,7 +135,7 @@ class SKLearnDatasets(BaseDynamicWrapperTemplate):
     @staticmethod
     def split_dataset(
         results: pd.DataFrame, feature_name_cols: list, target_name_cols: list, n_features: int, split_size: float
-    ) -> TabularDatasetSplit:
+    ) -> dict:
         """Method to split the dataset into training and testing samples"""
         if feature_name_cols:
             X = results[feature_name_cols]
@@ -154,7 +154,7 @@ class SKLearnDatasets(BaseDynamicWrapperTemplate):
             y_test=pd.DataFrame(y_test),
         )
 
-        return split_data
+        return split_data.model_dump()
 
     def execute(self, container: DataContainer) -> DataContainer:
         sklearn_dataset = self.wrapped_callable.__func__(**self.dataset_attributes.model_dump())

{sinapsis_data_readers-0.1.16 → sinapsis_data_readers-0.1.17}/src/sinapsis_data_readers/templates/datasets_readers/sktime_datasets.py

@@ -94,7 +94,7 @@ class SKTimeDatasets(BaseDynamicWrapperTemplate):
 
     def initialize_attributes(self):
         return getattr(self.attributes, self.wrapped_callable.__name__)
-    def split_time_series_dataset(self, dataset: Any) -> TabularDatasetSplit:
+    def split_time_series_dataset(self, dataset: Any) -> dict:
         """Split a time series dataset into training and testing sets
 
         Args:
@@ -104,12 +104,13 @@ class SKTimeDatasets(BaseDynamicWrapperTemplate):
             TabularDatasetSplit: Object containing the split time series data
         """
         y_train, y_test = temporal_train_test_split(dataset, train_size=self.attributes.train_size)
-        return TabularDatasetSplit(
+        split_data =  TabularDatasetSplit(
             x_train=pd.DataFrame(index=y_train.index),
             x_test=pd.DataFrame(index=y_test.index),
             y_train=pd.DataFrame(y_train),
             y_test=pd.DataFrame(y_test),
         )
+        return split_data.model_dump()
 
     def split_classification_dataset(self, X: Any, y: Any) -> TabularDatasetSplit:
         """Split a classification dataset into training and testing sets

{sinapsis_data_readers-0.1.16 → sinapsis_data_readers-0.1.17}/src/sinapsis_data_readers/templates/image_readers/coco_dataset_reader.py

@@ -55,8 +55,7 @@ class CocoImageDatasetBaseCV2(FolderImageDatasetCV2):
         annotations_path: str
 
     def __init__(self, attributes: TemplateAttributeType) -> None:
-        self.annotations_file = os.path.join(attributes.get("root_dir", SINAPSIS_CACHE_DIR),
-                                             attributes.get("data_dir"), attributes.get("annotations_path"))
+        self.annotations_file = os.path.join(attributes.get("root_dir", SINAPSIS_CACHE_DIR), attributes.get("data_dir"), attributes.get("annotations_path"))
         self.raw_annotations_dict: list[dict[str, dict[str, Any]]] = self.read_annotations_file(self.annotations_file)
         self.annotations = self.images_annotations()
         super().__init__(attributes)

{sinapsis_data_readers-0.1.16 → sinapsis_data_readers-0.1.17/src/sinapsis_data_readers.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-data-readers
-Version: 0.1.16
+Version: 0.1.17
 Summary: Templates to read data in different formats
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech

{sinapsis_data_readers-0.1.16 → sinapsis_data_readers-0.1.17}/src/sinapsis_data_readers.egg-info/SOURCES.txt

@@ -12,7 +12,6 @@ src/sinapsis_data_readers/helpers/__init__.py
 src/sinapsis_data_readers/helpers/coco_dataclasses.py
 src/sinapsis_data_readers/helpers/csv_reader.py
 src/sinapsis_data_readers/helpers/file_path_helpers.py
-src/sinapsis_data_readers/helpers/image_color_space_converter.py
 src/sinapsis_data_readers/helpers/sklearn_dataset_subset.py
 src/sinapsis_data_readers/helpers/sktime_datasets_subset.py
 src/sinapsis_data_readers/helpers/tags.py

sinapsis_data_readers-0.1.16/src/sinapsis_data_readers/helpers/image_color_space_converter.py

@@ -1,53 +0,0 @@
-# -*- coding: utf-8 -*-
-from enum import Enum
-
-import cv2
-from sinapsis_core.data_containers.data_packet import ImageColor, ImagePacket
-from sinapsis_core.utils.logging_utils import sinapsis_logger
-
-color_mapping = {
-    (ImageColor.RGB, ImageColor.BGR): cv2.COLOR_RGB2BGR,
-    (ImageColor.RGB, ImageColor.GRAY): cv2.COLOR_RGB2GRAY,
-    (ImageColor.RGB, ImageColor.RGBA): cv2.COLOR_RGB2RGBA,
-    (ImageColor.BGR, ImageColor.RGB): cv2.COLOR_BGR2RGB,
-    (ImageColor.BGR, ImageColor.GRAY): cv2.COLOR_BGR2GRAY,
-    (ImageColor.BGR, ImageColor.RGBA): cv2.COLOR_BGR2RGBA,
-    (ImageColor.GRAY, ImageColor.RGB): cv2.COLOR_GRAY2RGB,
-    (ImageColor.GRAY, ImageColor.BGR): cv2.COLOR_GRAY2BGR,
-    (ImageColor.GRAY, ImageColor.RGBA): cv2.COLOR_GRAY2RGBA,
-    (ImageColor.RGBA, ImageColor.RGB): cv2.COLOR_RGBA2RGB,
-    (ImageColor.RGBA, ImageColor.BGR): cv2.COLOR_RGBA2BGR,
-    (ImageColor.RGBA, ImageColor.GRAY): cv2.COLOR_RGBA2GRAY,
-}
-
-
-def convert_color_space_cv(image: ImagePacket, desired_color_space: Enum) -> ImagePacket:
-    """Converts an image from one color space to another, provided
-        they are in the color mapping options.
-
-    Args:
-        image (ImagePacket): Image packet to apply the conversion
-        desired_color_space (Enum): Color space to convert the image
-
-    Returns:
-        ImagePacket: Updated ImagePacket with content converted into the new color space
-        
-    Raises:
-         ValueError: If the conversion is not possible, return an error.
-
-    """
-    current_color_space = image.color_space
-
-    if (current_color_space, desired_color_space) in color_mapping:
-        conversion_code = color_mapping[(current_color_space, desired_color_space)]
-        try:
-            image.content = cv2.cvtColor(image.content, conversion_code)
-            image.color_space = desired_color_space
-
-        except cv2.error:
-            sinapsis_logger.error(f"Invalid conversion between {current_color_space} and {desired_color_space}")
-
-    else:
-        raise ValueError(f"Conversion from {current_color_space} to {desired_color_space} is not supported.")
-
-    return image