isolate changes to the postgres method

vanna-ai · Aug 21, 2024 · 78c7efc · 78c7efc
1 parent dad1f41
commit 78c7efc
Showing 1 changed file with 4 additions and 10 deletions.
diff --git a/src/vanna/base/base.py b/src/vanna/base/base.py
@@ -464,7 +464,7 @@ def get_training_data(self, **kwargs) -> pd.DataFrame:
         pass
 
     @abstractmethod
-    def remove_training_data(id: str, **kwargs) -> bool:
+    def remove_training_data(self, id: str, **kwargs) -> bool:
         """
         Example:
         ```python
@@ -1322,15 +1322,9 @@ def connect_to_bigquery(
 
         def run_sql_bigquery(sql: str) -> Union[pd.DataFrame, None]:
             if conn:
-                try:
-                    job = conn.query(sql)
-                    df = job.result().to_dataframe()
-                    return df
-                except GoogleAPIError as error:
-                    errors = []
-                    for error in error.errors:
-                        errors.append(error["message"])
-                    raise errors
+                job = conn.query(sql)
+                df = job.result().to_dataframe()
+                return df
             return None
 
         self.dialect = "BigQuery SQL"