diff --git a/python/mlcroissant/mlcroissant/_src/operation_graph/operations/field.py b/python/mlcroissant/mlcroissant/_src/operation_graph/operations/field.py index 71b3e464..5fd112db 100644 --- a/python/mlcroissant/mlcroissant/_src/operation_graph/operations/field.py +++ b/python/mlcroissant/mlcroissant/_src/operation_graph/operations/field.py @@ -230,9 +230,8 @@ def _get_result(row): value = None elif is_repeated: value = [ - _cast_value(self.node.ctx, v, field.data_type) - for v in value - ] + _cast_value(self.node.ctx, v, field.data_type) for v in value + ] else: value = _cast_value(self.node.ctx, value, field.data_type)