fix test

pola-rs · Aug 5, 2024 · a9e4cec · a9e4cec
1 parent 8898011
commit a9e4cec
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/py-polars/polars/io/spreadsheet/_write_utils.py b/py-polars/polars/io/spreadsheet/_write_utils.py
@@ -396,7 +396,11 @@ def _map_str(s: Series) -> Series:
                 df, row_totals, expand_keys=False, expand_values=True
             )
             row_totals_dtype = {  # type: ignore[assignment]
-                nm: (Int64 if _all_integer_cols(cols, schema) else Float64)
+                nm: (
+                    Int64
+                    if _all_integer_cols(numeric_cols if cols is True else cols, schema)
+                    else Float64
+                )
                 for nm, cols in row_totals.items()
             }
             row_total_funcs = {