bugfix in ts dataloader (#235)

* remove concat from time series dataloader unpack * pin xgboost
vanderschaarlab · Sep 12, 2023 · d36f37f · d36f37f
1 parent 7c1d9b5
commit d36f37f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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(
                     [