updated the smolagents

Oracle/DataSmolAgent.py

@@ -5,8 +5,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 
 @tool
 def clean_data(df: pd.DataFrame) -> pd.DataFrame:
-    """
-    Clean the DataFrame by stripping whitespace from column names and dropping rows that are completely empty.
+    """Clean the DataFrame by stripping whitespace from column names and dropping rows that are completely empty.
 
     Args:
         df (pd.DataFrame): The input DataFrame containing the raw data.
@@ -20,12 +19,11 @@ def clean_data(df: pd.DataFrame) -> pd.DataFrame:
 
 @tool
 def extract_features(df: pd.DataFrame) -> pd.DataFrame:
-    """
-    Dynamically extract features from the DataFrame.
-    
+    """Dynamically extract features from the DataFrame.
+
     Args:
         df (pd.DataFrame): The input DataFrame containing the raw data.
-        
+
     Returns:
         pd.DataFrame: The DataFrame updated with new dynamically engineered features.
     """
@@ -61,12 +59,11 @@ def extract_features(df: pd.DataFrame) -> pd.DataFrame:
 
 @tool
 def save_to_csv(df: pd.DataFrame, filename: str = "output.csv") -> str:
-    """
-    Save the DataFrame to a CSV file.
+    """Save the DataFrame to a CSV file.
 
     Args:
         df (pd.DataFrame): The DataFrame to save.
-        filename (str): The name of the output CSV file.
+        filename (str): The name of the output CSV file (default "output.csv").
 
     Returns:
         str: The file path of the saved CSV.