minor changes

Author: matteocacciola

pandasai/agent/base.py

@@ -175,6 +175,7 @@ def generate_code_with_retries(self, query: str) -> Any:
                     self._state.logger.log(
                         f"Retrying Code Generation ({attempts}/{max_retries})..."
                     )
+            return None
 
     def execute_with_retries(self, code: str) -> Any:
         """Execute the code with retry logic."""
@@ -194,6 +195,7 @@ def execute_with_retries(self, code: str) -> Any:
                     f"Retrying execution ({attempts}/{max_retries})..."
                 )
                 code = self._regenerate_code_after_error(code, e)
+        return None
 
     def train(
         self,