update pandasai/smart_dataframe/__init__.py

sinaptik-ai · Feb 6, 2024 · 8d6d069 · 8d6d069
1 parent eae3ca9
commit 8d6d069
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pandasai/smart_dataframe/__init__.py b/pandasai/smart_dataframe/__init__.py
@@ -391,8 +391,7 @@ def _truncate_head_columns(self, df: DataFrameType, max_size=25) -> DataFrameTyp
         Returns:
             DataFrameType: Pandas, Modin or Polars dataframe
         """
-        engine = df_type(df)
-        if engine in ("pandas", "modin"):
+        if (engine := df_type(df)) in ("pandas", "modin"):
             df_trunc = df.copy()
 
             for col in df.columns: