fix pandas df in update

capitalone · Mar 24, 2024 · f149792 · f149792
1 parent 16575a2
commit f149792
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/dataprofiler/profilers/order_column_profile.py b/dataprofiler/profilers/order_column_profile.py
@@ -4,6 +4,7 @@
 from abc import abstractmethod
 from typing import Protocol, Type, TypeVar, cast
 
+import pandas as pd
 import polars as pl
 from polars import Series
 
@@ -482,6 +483,9 @@ def update(self, df_series: Series) -> OrderColumn:
         :return: updated OrderColumn
         :rtype: OrderColumn
         """
+        # TODO remove onces profiler builder is updated
+        if type(df_series) == pd.Series:
+            df_series = pl.from_pandas(df_series)  # type: ignore
         if len(df_series) == 0:
             return self