feat(eda): support series as the input

Author: dovahcrow

dataprep/eda/data_array.py

@@ -186,6 +186,9 @@ class DataArray:
     def __init__(
         self, df: DataFrame, value_length: bool = False, repartition: bool = True,
     ) -> None:
+        if isinstance(df, (dd.Series, pd.Series)):
+            df = df.to_frame()
+
         # numpy does not understand pandas types
         cat_cols = [
             col

dataprep/eda/distribution/compute/__init__.py

@@ -94,8 +94,8 @@ def compute(
         or dtype = Continuous() or dtype = "Continuous" or dtype = Continuous()
     """  # pylint: disable=too-many-locals
 
-    df.columns = df.columns.astype(str)
     df = to_dask(df)
+    df.columns = df.columns.astype(str)
 
     if not any((x, y, z)):
         return compute_overview(df, bins, ngroups, largest, timeunit, dtype)

dataprep/eda/utils.py

@@ -35,6 +35,11 @@ def to_dask(df: Union[pd.DataFrame, dd.DataFrame]) -> dd.DataFrame:
     """Convert a dataframe to a dask dataframe."""
     if isinstance(df, dd.DataFrame):
         return df
+    elif isinstance(df, dd.Series):
+        return df.to_frame()
+
+    if isinstance(df, pd.Series):
+        df = df.to_frame()
 
     df_size = df.memory_usage(deep=True).sum()
     npartitions = ceil(df_size / 128 / 1024 / 1024)  # 128 MB partition size