fix: input dataframe prep col types

GoogleCloudPlatform · Aug 27, 2024 · d772bb5 · d772bb5
1 parent afc0d0c
commit d772bb5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/dfcx_scrapi/tools/evaluations.py b/src/dfcx_scrapi/tools/evaluations.py
@@ -513,9 +513,8 @@ def validate_and_prep_inputs(self, df: pd.DataFrame) -> pd.DataFrame:
 
         # fill remaining NA with empty string
         for col in df.columns:
-            if df[col].dtype == "object":
+            if df[col].dtype in ["object", "string"]:
                 df[col] = df[col].fillna("")
-        # df.fillna("", inplace=True)
 
         return df