fix(eda): change dtype 'string' to 'object'

sfu-db · Sep 30, 2020 · aaa0df2 · aaa0df2
1 parent df32e1d
commit aaa0df2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/dataprep/eda/distribution/compute/__init__.py b/dataprep/eda/distribution/compute/__init__.py
@@ -93,6 +93,10 @@ def compute(
         dtype = {"a": Continuous(), "b": "nominal"}
         or dtype = Continuous() or dtype = "Continuous" or dtype = Continuous()
     """  # pylint: disable=too-many-locals
+
+    for col in df:
+        if df[col].dtype=='string':
+            df[col] = df[col].astype('object') 
 
     df = to_dask(df)
     df.columns = df.columns.astype(str)