fix(connector): pull dataframe fixed

pandasai/data_loader/loader.py

@@ -2,7 +2,6 @@
 import hashlib
 import importlib
 import os
-from datetime import datetime, timedelta
 from typing import Any, Optional
 
 import pandas as pd

pandasai/dataframe/base.py

@@ -237,7 +237,12 @@ def pull(self):
         dataset_loader = DatasetLoader()
         df = dataset_loader.load(self.path)
         self.__init__(
-            df, schema=df.schema, name=df.name, description=df.description, path=df.path
+            df,
+            schema=df.schema,
+            data_loader=dataset_loader,
+            name=df.name,
+            description=df.description,
+            path=df.path,
         )
 
         print(f"Dataset pulled successfully from path: {self.path}")

pandasai/dataframe/virtual_dataframe.py

@@ -36,14 +36,18 @@ def __init__(self, *args, **kwargs):
         if not schema:
             raise VirtualizationError("Schema is required for virtualization!")
 
-        table_name = schema.source.table or schema.name
+        name = kwargs.pop("name", None)
 
-        description = schema.description
+        description = kwargs.pop("description", None)
+
+        table_name = schema.source.table or name or schema.name
+
+        table_description = description or schema.description
 
         super().__init__(
             self.get_head(),
             name=table_name,
-            description=description,
+            description=table_description,
             *args,
             **kwargs,
         )