sinapsis-data-readers 0.1.19__py3-none-any.whl → 0.1.20__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -38,7 +38,11 @@ class ImageDatasetSplit(BaseModel):
38
38
  """allow arbitrary types"""
39
39
 
40
40
  arbitrary_types_allowed = True
41
-
41
+ json_encoders : dict = {
42
+ pd.DataFrame: lambda df: df.to_dict(orient="records"),
43
+ pd.Series: lambda s: s.to_list(),
44
+ np.ndarray: lambda arr: arr.tolist(),
45
+ }
42
46
 
43
47
  class TabularDatasetSplit(BaseModel):
44
48
  """BaseModel to store the content of the data packets as a list
@@ -126,15 +126,17 @@ class SKTimeDatasets(BaseDynamicWrapperTemplate):
126
126
  X_train, X_test, y_train, y_test = train_test_split(
127
127
  X, y, train_size=self.attributes.train_size, random_state=0
128
128
  )
129
- return TabularDatasetSplit(
129
+ split_dataset = TabularDatasetSplit(
130
130
  x_train=pd.DataFrame(X_train),
131
131
  x_test=pd.DataFrame(X_test),
132
132
  y_train=pd.DataFrame(y_train),
133
133
  y_test=pd.DataFrame(y_test),
134
134
  )
135
+ return split_dataset.model_dump_json(indent=2)
135
136
  except ValueError:
136
137
  self.logger.debug("Wrong format for split. original values")
137
- return TabularDatasetSplit(x_train=pd.DataFrame(X), y_train=pd.DataFrame(y))
138
+ split_dataset = TabularDatasetSplit(x_train=pd.DataFrame(X), y_train=pd.DataFrame(y))
139
+ return split_dataset.model_dump_json(indent=2)
138
140
 
139
141
  def create_dataset(self):
140
142
  return self.wrapped_callable.__func__(**self.dataset_attributes.model_dump())
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: sinapsis-data-readers
3
- Version: 0.1.19
3
+ Version: 0.1.20
4
4
  Summary: Templates to read data in different formats
5
5
  Author-email: SinapsisAI <dev@sinapsis.tech>
6
6
  Project-URL: Homepage, https://sinapsis.tech
@@ -17,9 +17,9 @@ sinapsis_data_readers/templates/audio_readers/audio_reader_to_bytes.py,sha256=ex
17
17
  sinapsis_data_readers/templates/audio_readers/base_audio_reader.py,sha256=-3biMclztIxDCwPaQq4kUkcY2BaP13saxWdIQdm-e-E,2284
18
18
  sinapsis_data_readers/templates/datasets_readers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
19
19
  sinapsis_data_readers/templates/datasets_readers/csv_datasets.py,sha256=jn2x8QXpDn-wLoXML24xcfJ29WHT2OhRuFNrU105lOs,1210
20
- sinapsis_data_readers/templates/datasets_readers/dataset_splitter.py,sha256=6FqN1x6V748Q_ESFfxfaCRJKJQY8cK-gwRHvzuYdVqI,8860
20
+ sinapsis_data_readers/templates/datasets_readers/dataset_splitter.py,sha256=lnOfpFCl_ozp1EcYkHL77pJugA1TDEGu8m4Dojzyyuc,9065
21
21
  sinapsis_data_readers/templates/datasets_readers/sklearn_datasets.py,sha256=Nr3RjdXxjtRz-mf2SwLDlpZ_AvWsno476n0j9GuLaOc,8244
22
- sinapsis_data_readers/templates/datasets_readers/sktime_datasets.py,sha256=TEUQVBP7sCYh6hvCX3W1N6c3Cn_v012dfaFKzJFqek4,9500
22
+ sinapsis_data_readers/templates/datasets_readers/sktime_datasets.py,sha256=RJ8hYVdqxfciftMLj3gha3It8GXK2mMlu0CyjKJvkcg,9636
23
23
  sinapsis_data_readers/templates/image_readers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
24
24
  sinapsis_data_readers/templates/image_readers/base_image_folder_data_loader.py,sha256=UM2SmBX7B2RE1OgwRtBn-GZdKmrgZILdN32itHIJKQI,4832
25
25
  sinapsis_data_readers/templates/image_readers/coco_dataset_reader.py,sha256=cFVP50LeRSu9tppy4RYZbAmN775f6AuGUdDTzAHRf3k,14675
@@ -34,8 +34,8 @@ sinapsis_data_readers/templates/video_readers/video_reader_cv2.py,sha256=n3EYBYg
34
34
  sinapsis_data_readers/templates/video_readers/video_reader_dali.py,sha256=TIsV1h4eYVOocdwC8tqW3aYz_9l5XJq5nTVc4rWlFUQ,8804
35
35
  sinapsis_data_readers/templates/video_readers/video_reader_ffmpeg.py,sha256=uAnV02i9gy7p9mxTVcx20F6ily4JhBtJSDJ93Reyi4w,5046
36
36
  sinapsis_data_readers/templates/video_readers/video_reader_torchcodec.py,sha256=wMCjRCaMknDYPOsspuf7NVEhy49h818lVq__AHTsFA8,4036
37
- sinapsis_data_readers-0.1.19.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
38
- sinapsis_data_readers-0.1.19.dist-info/METADATA,sha256=BAgUowN5EKwL35zhcJQB-O3YazivmRbiEpDtRBGWALs,6520
39
- sinapsis_data_readers-0.1.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
40
- sinapsis_data_readers-0.1.19.dist-info/top_level.txt,sha256=3R3oDiABqDVBW2Fc-SpWXYHkRdYFGZjT_wo6Q0Uqnhw,41
41
- sinapsis_data_readers-0.1.19.dist-info/RECORD,,
37
+ sinapsis_data_readers-0.1.20.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
38
+ sinapsis_data_readers-0.1.20.dist-info/METADATA,sha256=vBwNgywxw-uOx5se27Xm4xKKDDbAIjr2LCM0LkNyeYk,6520
39
+ sinapsis_data_readers-0.1.20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
40
+ sinapsis_data_readers-0.1.20.dist-info/top_level.txt,sha256=3R3oDiABqDVBW2Fc-SpWXYHkRdYFGZjT_wo6Q0Uqnhw,41
41
+ sinapsis_data_readers-0.1.20.dist-info/RECORD,,