Skip to content

Commit 6f1ee74

Browse files
committed
stringify list columns better brackets now #97
1 parent b7c1ae4 commit 6f1ee74

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

file_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def _stringify_list_columns(df: pl.DataFrame) -> pl.DataFrame:
1616
list_cols = [col for col, dtype in zip(df.columns, df.dtypes) if dtype.base_type() == pl.List]
1717
if list_cols:
1818
df = df.with_columns(
19-
[pl.col(c).map_elements(lambda val: json.dumps(val, default=str), return_dtype=pl.String).alias(c) for c in list_cols]
19+
[pl.col(c).map_elements(lambda val: json.dumps(val.to_list(), default=str), return_dtype=pl.String).alias(c) for c in list_cols]
2020
)
2121
return df
2222

0 commit comments

Comments
 (0)