bugfix in ts dataloader (#235)

robsdavis · web-flow · commit d36f37fcd36d · 2023-09-12T19:05:01.000+01:00
* remove concat from time series dataloader unpack

* pin xgboost
diff --git a/setup.cfg b/setup.cfg
@@ -47,7 +47,7 @@ install_requires =
     pydantic<2.0
     cloudpickle
     scipy
-    xgboost
+    xgboost<2.0.0
     geomloss
     pgmpy
     redis
diff --git a/src/synthcity/plugins/core/dataloader.py b/src/synthcity/plugins/core/dataloader.py
@@ -932,7 +932,7 @@ def unpack(self, as_numpy: bool = False, pad: bool = False) -> Any:
             longest_observation_seq = max([len(seq) for seq in temporal_data])
             return (
                 np.asarray(static_data),
-                np.asarray(pd.concat(temporal_data)),
+                np.asarray(temporal_data),
                 # masked array to handle variable length sequences
                 ma.vstack(
                     [

Original file line number	Diff line number	Diff line change
`@@ -932,7 +932,7 @@ def unpack(self, as_numpy: bool = False, pad: bool = False) -> Any:`
`932`	`932`	`longest_observation_seq = max([len(seq) for seq in temporal_data])`
`933`	`933`	`return (`
`934`	`934`	`np.asarray(static_data),`
`935`		`- np.asarray(pd.concat(temporal_data)),`
	`935`	`+ np.asarray(temporal_data),`
`936`	`936`	`# masked array to handle variable length sequences`
`937`	`937`	`ma.vstack(`
`938`	`938`	`[`