Update field.py

To solve bad return type error
mlcommons · Nov 3, 2023 · 39877c9 · 39877c9
1 parent eb48502
commit 39877c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/mlcroissant/mlcroissant/_src/operation_graph/operations/field.py b/python/mlcroissant/mlcroissant/_src/operation_graph/operations/field.py
@@ -118,4 +118,4 @@ def __call__(self, df: pd.DataFrame) -> pd.Series:
         series = _name_series(value, self.node)
         transforms = functools.partial(apply_transforms_fn, source=self.node.source)
         cast = functools.partial(_cast_value, data_type=self.node.data_type)
-        return series.apply(transforms).apply(cast)
+        return pd.Series(series.apply(transforms).apply(cast))