Dont convert bool columns (To add string column)

BackofenLab · Nov 2, 2023 · c3347e2 · c3347e2
1 parent 4cb24ee
commit c3347e2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/db/scripts/utils.py b/db/scripts/utils.py
@@ -285,7 +285,9 @@ def get_values_reformat(df: pd.DataFrame, match: list):
     """
     values = list(set(list(df.columns)) - set(match))
     reformat = [
-        (i, "toFloat" if df[i].dtype == "float64" else "toInteger") for i in list(df.columns) if df[i].dtype != "object"
+        (i, "toFloat" if df[i].dtype == "float64" else "toInteger")
+        for i in list(df.columns)
+        if df[i].dtype != "object" and df[i].dtype != "bool"
     ]
     return values, reformat