🎨 Changed to map and added super() in to_df method

dyvenia · Oct 4, 2024 · c74367f · c74367f
1 parent 735311d
commit c74367f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/viadot/sources/azure_sql.py b/src/viadot/sources/azure_sql.py
@@ -154,7 +154,7 @@ def to_df(
                 remove_special_characters. If None whole data frame will be processed.
                 Defaults to None.
         """
-        df = self.to_df(query=query, if_empty=if_empty)
+        df = super().to_df(query=query, if_empty=if_empty)
 
         if convert_bytes:
             df = df_converts_bytes_to_int(df=df)

diff --git a/src/viadot/utils.py b/src/viadot/utils.py
@@ -973,7 +973,7 @@ def df_converts_bytes_to_int(df: pd.DataFrame) -> pd.DataFrame:
     Returns:
         pd.DataFrame: Data Frame after convert
     """
-    return df.applymap(lambda x: int(x) if isinstance(x, bytes) else x)
+    return df.map(lambda x: int(x) if isinstance(x, bytes) else x)
 
 
 def df_clean_column(