reseting index of dataframes after transformation (logicalclocks#252)

manu-sj · web-flow · commit 5a1b4dfadd58 · 2024-07-31T15:27:56.000+02:00
diff --git a/python/hsfs/engine/python.py b/python/hsfs/engine/python.py
@@ -1236,15 +1236,15 @@ def _apply_transformation_function(
                 dropped_features.update(tf.hopsworks_udf.dropped_features)
             dataset = pd.concat(
                 [
-                    dataset,
+                    dataset.reset_index(drop=True),
                     tf.hopsworks_udf.get_udf()(
                         *(
                             [
                                 dataset[feature]
                                 for feature in tf.hopsworks_udf.transformation_features
                             ]
                         )
-                    ),
+                    ).reset_index(drop=True),
                 ],
                 axis=1,
             )

Original file line number	Diff line number	Diff line change
`@@ -1236,15 +1236,15 @@ def _apply_transformation_function(`
`1236`	`1236`	`dropped_features.update(tf.hopsworks_udf.dropped_features)`
`1237`	`1237`	`dataset = pd.concat(`
`1238`	`1238`	`[`
`1239`		`- dataset,`
	`1239`	`+ dataset.reset_index(drop=True),`
`1240`	`1240`	`tf.hopsworks_udf.get_udf()(`
`1241`	`1241`	`*(`
`1242`	`1242`	`[`
`1243`	`1243`	`dataset[feature]`
`1244`	`1244`	`for feature in tf.hopsworks_udf.transformation_features`
`1245`	`1245`	`]`
`1246`	`1246`	`)`
`1247`		`- ),`
	`1247`	`+ ).reset_index(drop=True),`
`1248`	`1248`	`],`
`1249`	`1249`	`axis=1,`
`1250`	`1250`	`)`