Fix flake

mlcommons · Dec 2, 2024 · 0b5cdfd · 0b5cdfd
1 parent a9a5767
commit 0b5cdfd
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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)