sinapsis-data-tools 0.2.20__py3-none-any.whl → 0.2.23__py3-none-any.whl

sinapsis_data_readers/src/sinapsis_data_readers/templates/datasets_readers/dataset_splitter.py

@@ -38,11 +38,7 @@ class ImageDatasetSplit(BaseModel):
         """allow arbitrary types"""
 
         arbitrary_types_allowed = True
-        json_encoders : dict = {
-            pd.DataFrame: lambda df: df.to_dict(orient="records"),
-            pd.Series: lambda s: s.to_list(),
-            np.ndarray: lambda arr: arr.tolist(),
-        }
+
 
 class TabularDatasetSplit(BaseModel):
     """BaseModel to store the content of the data packets as a list
@@ -65,6 +61,11 @@ class TabularDatasetSplit(BaseModel):
         """allow arbitrary types"""
 
         arbitrary_types_allowed = True
+        json_encoders: dict = {
+            pd.DataFrame: lambda df: df.to_dict(orient="records"),
+            pd.Series: lambda s: s.to_list(),
+            np.ndarray: lambda arr: arr.tolist(),
+        }
 
 
 class DatasetSplitterBase(Template):

sinapsis_data_readers/src/sinapsis_data_readers/templates/datasets_readers/sktime_datasets.py

@@ -126,15 +126,17 @@ class SKTimeDatasets(BaseDynamicWrapperTemplate):
             X_train, X_test, y_train, y_test = train_test_split(
                 X, y, train_size=self.attributes.train_size, random_state=0
             )
-            return TabularDatasetSplit(
+            split_dataset = TabularDatasetSplit(
                 x_train=pd.DataFrame(X_train),
                 x_test=pd.DataFrame(X_test),
                 y_train=pd.DataFrame(y_train),
                 y_test=pd.DataFrame(y_test),
             )
+            return split_dataset.model_dump_json(indent=2)
         except ValueError:
             self.logger.debug("Wrong format for split. original values")
-            return TabularDatasetSplit(x_train=pd.DataFrame(X), y_train=pd.DataFrame(y))
+            split_dataset = TabularDatasetSplit(x_train=pd.DataFrame(X), y_train=pd.DataFrame(y))
+            return split_dataset.model_dump_json(indent=2)
 
     def create_dataset(self):
         return self.wrapped_callable.__func__(**self.dataset_attributes.model_dump())

{sinapsis_data_tools-0.2.20.dist-info → sinapsis_data_tools-0.2.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-data-tools
-Version: 0.2.20
+Version: 0.2.23
 Summary: Module that contains different packages to perform data-related operations through Sinapsis templates.
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech
@@ -19,7 +19,7 @@ Requires-Dist: sinapsis-data-analysis; extra == "all"
 Requires-Dist: sinapsis-data-writers[all]; extra == "all"
 Requires-Dist: sinapsis-data-readers[all]; extra == "all"
 Requires-Dist: sinapsis-data-visualization[all]; extra == "all"
-Requires-Dist: sinapsis-generic-data-tools; extra == "all"
+Requires-Dist: sinapsis-generic-data-tools[all]; extra == "all"
 Dynamic: license-file
 
 <h1 align="center">

{sinapsis_data_tools-0.2.20.dist-info → sinapsis_data_tools-0.2.23.dist-info}/RECORD

@@ -30,9 +30,9 @@ sinapsis_data_readers/src/sinapsis_data_readers/templates/audio_readers/audio_re
 sinapsis_data_readers/src/sinapsis_data_readers/templates/audio_readers/base_audio_reader.py,sha256=-3biMclztIxDCwPaQq4kUkcY2BaP13saxWdIQdm-e-E,2284
 sinapsis_data_readers/src/sinapsis_data_readers/templates/datasets_readers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_data_readers/src/sinapsis_data_readers/templates/datasets_readers/csv_datasets.py,sha256=jn2x8QXpDn-wLoXML24xcfJ29WHT2OhRuFNrU105lOs,1210
-sinapsis_data_readers/src/sinapsis_data_readers/templates/datasets_readers/dataset_splitter.py,sha256=lnOfpFCl_ozp1EcYkHL77pJugA1TDEGu8m4Dojzyyuc,9065
+sinapsis_data_readers/src/sinapsis_data_readers/templates/datasets_readers/dataset_splitter.py,sha256=mhllO_2aBwQbxP1wr9G_RAzb6KffX1In0d-BLp-BhFM,9065
 sinapsis_data_readers/src/sinapsis_data_readers/templates/datasets_readers/sklearn_datasets.py,sha256=Nr3RjdXxjtRz-mf2SwLDlpZ_AvWsno476n0j9GuLaOc,8244
-sinapsis_data_readers/src/sinapsis_data_readers/templates/datasets_readers/sktime_datasets.py,sha256=TEUQVBP7sCYh6hvCX3W1N6c3Cn_v012dfaFKzJFqek4,9500
+sinapsis_data_readers/src/sinapsis_data_readers/templates/datasets_readers/sktime_datasets.py,sha256=RJ8hYVdqxfciftMLj3gha3It8GXK2mMlu0CyjKJvkcg,9636
 sinapsis_data_readers/src/sinapsis_data_readers/templates/image_readers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_data_readers/src/sinapsis_data_readers/templates/image_readers/base_image_folder_data_loader.py,sha256=UM2SmBX7B2RE1OgwRtBn-GZdKmrgZILdN32itHIJKQI,4832
 sinapsis_data_readers/src/sinapsis_data_readers/templates/image_readers/coco_dataset_reader.py,sha256=cFVP50LeRSu9tppy4RYZbAmN775f6AuGUdDTzAHRf3k,14675
@@ -47,7 +47,7 @@ sinapsis_data_readers/src/sinapsis_data_readers/templates/video_readers/video_re
 sinapsis_data_readers/src/sinapsis_data_readers/templates/video_readers/video_reader_dali.py,sha256=TIsV1h4eYVOocdwC8tqW3aYz_9l5XJq5nTVc4rWlFUQ,8804
 sinapsis_data_readers/src/sinapsis_data_readers/templates/video_readers/video_reader_ffmpeg.py,sha256=uAnV02i9gy7p9mxTVcx20F6ily4JhBtJSDJ93Reyi4w,5046
 sinapsis_data_readers/src/sinapsis_data_readers/templates/video_readers/video_reader_torchcodec.py,sha256=wMCjRCaMknDYPOsspuf7NVEhy49h818lVq__AHTsFA8,4036
-sinapsis_data_tools-0.2.20.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+sinapsis_data_tools-0.2.23.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
 sinapsis_data_visualization/src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_data_visualization/src/sinapsis_data_visualization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_data_visualization/src/sinapsis_data_visualization/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -59,7 +59,7 @@ sinapsis_data_visualization/src/sinapsis_data_visualization/helpers/plot_distrib
 sinapsis_data_visualization/src/sinapsis_data_visualization/helpers/scikit_pca_analysis.py,sha256=dPma6VpJwluBub60o9p38xz_ohOnCLfzbCBDGb3eIVc,1702
 sinapsis_data_visualization/src/sinapsis_data_visualization/helpers/tags.py,sha256=tIOiyCuGujmbl0tNcVGKkwkzmXGZpnN2RydSNXPZask,702
 sinapsis_data_visualization/src/sinapsis_data_visualization/templates/__init__.py,sha256=0as-_35ugZRXk3WUwy84m2-NrZpj5BNBJlMLde6D7xo,943
-sinapsis_data_visualization/src/sinapsis_data_visualization/templates/base_annotation_drawer.py,sha256=F-JczMFwX6B0VmRj6qWU7RgPrkuE2z3cxkZLjc1G_aU,6042
+sinapsis_data_visualization/src/sinapsis_data_visualization/templates/base_annotation_drawer.py,sha256=dfwrsIDWfLqwrZVO_F_aHrDAw8dNDEYwrSgjv_wNQR8,6043
 sinapsis_data_visualization/src/sinapsis_data_visualization/templates/base_visualization_template.py,sha256=REAEU1Kwn93xSkevlqzZEGyXln7CVfnGAikr3fVSp2Y,8861
 sinapsis_data_visualization/src/sinapsis_data_visualization/templates/bbox_drawer.py,sha256=UdEDOytimBcyTNi8A4fcl8_-pk5M-hpnDK-S3gK_lic,3658
 sinapsis_data_visualization/src/sinapsis_data_visualization/templates/data_distribution_visualization.py,sha256=UqE6t3K5fAISgVbSK0CRpQ1b3Ge1S2NXkOGMntiiQEI,5270
@@ -99,7 +99,7 @@ sinapsis_generic_data_tools/src/sinapsis_generic_data_tools/templates/mask_non_r
 sinapsis_generic_data_tools/src/sinapsis_generic_data_tools/templates/packet_buffer_queue.py,sha256=xh1A4jgqwT09lfF3lQ2EcYvcJehYV0wlUc8iBNJPn9I,4494
 sinapsis_generic_data_tools/src/sinapsis_generic_data_tools/templates/source_history_aggregator.py,sha256=zvp75oBInYK5NoUEbuQkltizgtMfNFLZ7GpTUu5frCU,3220
 sinapsis_generic_data_tools/src/sinapsis_generic_data_tools/templates/text_content_filter.py,sha256=uglR0e4RAbSMdaWVMaPuwo9wtxR2z3w1XLyVOvZXwjU,2719
-sinapsis_data_tools-0.2.20.dist-info/METADATA,sha256=3pybI-heAomiWzcozUkZhLsjMDeyylhkMZmT0FcHNlo,7641
-sinapsis_data_tools-0.2.20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sinapsis_data_tools-0.2.20.dist-info/top_level.txt,sha256=XdkcGiHnbgb2Ad6RqbEAU_c_9n30eNTHUqvrDMXBQqE,123
-sinapsis_data_tools-0.2.20.dist-info/RECORD,,
+sinapsis_data_tools-0.2.23.dist-info/METADATA,sha256=IpHr9Tq_5P1ajULOo9WvIs_ridYIqhvaQbBAFwHJMJ4,7646
+sinapsis_data_tools-0.2.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sinapsis_data_tools-0.2.23.dist-info/top_level.txt,sha256=XdkcGiHnbgb2Ad6RqbEAU_c_9n30eNTHUqvrDMXBQqE,123
+sinapsis_data_tools-0.2.23.dist-info/RECORD,,

sinapsis_data_visualization/src/sinapsis_data_visualization/templates/base_annotation_drawer.py

@@ -169,6 +169,7 @@ class BaseAnnotationDrawer(Template, abc.ABC):
         Returns:
             DataContainer: The updated data container with images annotated.
         """
+
         self.process_images(container)
 
         return container