MindsDB 25.5.3.0__py3-none-any.whl → 25.5.4.0__py3-none-any.whl

mindsdb/integrations/handlers/github_handler/github_tables.py

@@ -1,14 +1,10 @@
 import re
 from typing import List
-
 import pandas as pd
 
-from mindsdb_sql_parser import ast
-
 from mindsdb.integrations.libs.api_handler import APIResource
-from mindsdb.integrations.utilities.handlers.query_utilities import SELECTQueryParser, SELECTQueryExecutor
 from mindsdb.integrations.utilities.sql_utils import (
-    extract_comparison_conditions, FilterCondition, FilterOperator, SortColumn)
+    FilterCondition, FilterOperator, SortColumn)
 from mindsdb.utilities import log
 
 
@@ -241,8 +237,6 @@ class GithubPullRequestsTable(APIResource):
              limit: int = None,
              sort: List[SortColumn] = None,
              targets: List[str] = None) -> pd.DataFrame:
-
-
         """Pulls data from the GitHub "List repository pull requests" API
 
         Native filters:
@@ -397,7 +391,6 @@ class GithubCommitsTable(APIResource):
              limit: int = None,
              sort: List[SortColumn] = None,
              targets: List[str] = None) -> pd.DataFrame:
-
         """Pulls data from the GitHub "List commits" API
 
         Returns
@@ -473,7 +466,6 @@ class GithubReleasesTable(APIResource):
              limit: int = None,
              sort: List[SortColumn] = None,
              targets: List[str] = None) -> pd.DataFrame:
-
         """Pulls data from the GitHub "List repository releases" API
 
         Returns
@@ -550,7 +542,6 @@ class GithubBranchesTable(APIResource):
              limit: int = None,
              sort: List[SortColumn] = None,
              targets: List[str] = None) -> pd.DataFrame:
-
         """Pulls data from the GitHub "List repository branches" API
 
         Returns
@@ -616,7 +607,6 @@ class GithubContributorsTable(APIResource):
              limit: int = None,
              sort: List[SortColumn] = None,
              targets: List[str] = None) -> pd.DataFrame:
-
         """Pulls data from the GitHub "List repository contributors" API
 
         Returns
@@ -718,7 +708,6 @@ class GithubProjectsTable(APIResource):
              limit: int = None,
              sort: List[SortColumn] = None,
              targets: List[str] = None) -> pd.DataFrame:
-
         """Pulls data from the GitHub "List repository projects" API
 
         Returns
@@ -799,6 +788,7 @@ class GithubProjectsTable(APIResource):
             "creator_site_admin"
         ]
 
+
 class GithubMilestonesTable(APIResource):
     """The GitHub Milestones Table implementation"""
 
@@ -807,7 +797,6 @@ class GithubMilestonesTable(APIResource):
              limit: int = None,
              sort: List[SortColumn] = None,
              targets: List[str] = None) -> pd.DataFrame:
-
         """Pulls data from the GitHub "List repository milestones" API
 
         Returns
@@ -899,17 +888,17 @@ class GithubFilesTable(APIResource):
                 limit -= len(subres)
             else:
                 if (
-                      (
+                    (
                         file_matches is None
-                        or
-                        any(re.match(pattern, item.name) for pattern in file_matches)
-                      )
-                    and
-                      (
-                          file_not_matches is None
-                          or
-                          not any(re.match(pattern, item.name) for pattern in file_not_matches)
-                      )
+
+                        or any(re.match(pattern, item.name) for pattern in file_matches)
+                    )
+
+                    and (
+                        file_not_matches is None
+
+                        or not any(re.match(pattern, item.name) for pattern in file_not_matches)
+                    )
                 ):
 
                     file = {
@@ -965,4 +954,4 @@ class GithubFilesTable(APIResource):
         return pd.DataFrame(res, columns=self.get_columns())
 
     def get_columns(self) -> list:
-        return ['path', 'name', 'content']
+        return ['path', 'name', 'content']

mindsdb/integrations/handlers/gitlab_handler/gitlab_handler.py

@@ -11,6 +11,7 @@ from mindsdb_sql_parser import parse_sql
 
 logger = log.getLogger(__name__)
 
+
 class GitlabHandler(APIHandler):
     """The GitLab handler implementation"""
 
@@ -60,7 +61,7 @@ class GitlabHandler(APIHandler):
         try:
             self.connect()
             if self.connection_data.get("api_key", None):
-                logger.info(f"Authenticated as user")
+                logger.info("Authenticated as user")
             else:
                 logger.info("Proceeding without an API key")
 

mindsdb/integrations/handlers/gitlab_handler/gitlab_tables.py

@@ -10,6 +10,7 @@ from mindsdb_sql_parser import ast
 
 logger = log.getLogger(__name__)
 
+
 class GitlabIssuesTable(APITable):
     """The GitLab Issue Table implementation"""
 
@@ -81,8 +82,6 @@ class GitlabIssuesTable(APITable):
 
         gitlab_issues_df = pd.DataFrame(columns=self.get_columns())
 
-        start = 0
-
         issues_kwargs["per_page"] = total_results
         issues_kwargs["get_all"] = False
         while True:
@@ -253,8 +252,6 @@ class GitlabMergeRequestsTable(APITable):
 
         gitlab_merge_requests_df = pd.DataFrame(columns=self.get_columns())
 
-        start = 0
-
         merge_requests_kwargs["per_page"] = total_results
         merge_requests_kwargs["get_all"] = False
         while True:

mindsdb/integrations/handlers/gmail_handler/gmail_handler.py

@@ -22,7 +22,7 @@ from email.message import EmailMessage
 
 from base64 import urlsafe_b64encode, urlsafe_b64decode
 
-from mindsdb.integrations.utilities.handlers.auth_utilities import GoogleUserOAuth2Manager
+from mindsdb.integrations.utilities.handlers.auth_utilities.google import GoogleUserOAuth2Manager
 from mindsdb.integrations.utilities.handlers.auth_utilities.exceptions import AuthException
 
 DEFAULT_SCOPES = [
@@ -59,20 +59,17 @@ class EmailsTable(APITable):
         conditions = extract_comparison_conditions(query.where)
 
         params = {}
-        include_attachments = False
         for op, arg1, arg2 in conditions:
 
             if op == 'or':
-                raise NotImplementedError(f'OR is not supported')
+                raise NotImplementedError('OR is not supported')
 
-            if arg1 in ['query', 'label_ids', 'include_spam_trash', 'include_attachments']:
+            if arg1 in ['query', 'label_ids', 'include_spam_trash']:
                 if op == '=':
                     if arg1 == 'query':
                         params['q'] = arg2
                     elif arg1 == 'label_ids':
                         params['labelIds'] = arg2.split(',')
-                    elif arg1 == 'include_attachments':
-                        include_attachments = arg2 == 'true'
                     else:
                         params['includeSpamTrash'] = arg2
                 else:
@@ -88,9 +85,6 @@ class EmailsTable(APITable):
             method_name='list_messages',
             params=params
         )
-        attachments = []
-        if include_attachments:
-            attachments = self.handler.get_attachments(result)
         # filter targets
         columns = []
         for target in query.targets:
@@ -172,7 +166,7 @@ class EmailsTable(APITable):
         for row in query.values:
             params = dict(zip(columns, row))
 
-            if not 'to_email' in params:
+            if 'to_email' not in params:
                 raise ValueError('"to_email" parameter is required to send an email')
 
             message = EmailMessage()
@@ -213,7 +207,7 @@ class EmailsTable(APITable):
         conditions = extract_comparison_conditions(query.where)
         for op, arg1, arg2 in conditions:
             if op == 'or':
-                raise NotImplementedError(f'OR is not supported')
+                raise NotImplementedError('OR is not supported')
             if arg1 == 'message_id':
                 if op == '=':
                     self.handler.call_gmail_api('delete_message', {'id': arg2})
@@ -235,7 +229,7 @@ class EmailsTable(APITable):
         conditions = extract_comparison_conditions(query.where)
         for op, arg1, arg2 in conditions:
             if op == 'or':
-                raise NotImplementedError(f'OR is not supported')
+                raise NotImplementedError('OR is not supported')
             if arg1 == 'id':
                 if op == '=':
                     params['id'] = arg2
@@ -434,7 +428,6 @@ class GmailHandler(APIHandler):
             for attachment in attachments:
                 attachment_id = attachment['attachmentId']
                 filename = attachment['filename']
-                mimeType = attachment['mimeType']
                 attachment_data = self.service.users().messages().attachments().get(
                     userId='me', messageId=email['id'], id=attachment_id).execute()
                 file_data = attachment_data['data']

mindsdb/integrations/handlers/google_books_handler/google_books_handler.py

@@ -16,6 +16,7 @@ from mindsdb.utilities import log
 
 logger = log.getLogger(__name__)
 
+
 class GoogleBooksHandler(APIHandler):
     """
         A class for handling connections and interactions with the Google Books API.
@@ -81,7 +82,7 @@ class GoogleBooksHandler(APIHandler):
         response = StatusResponse(False)
 
         try:
-            service = self.connect()
+            self.connect()
             response.success = True
         except Exception as e:
             logger.error(f'Error connecting to Google Books API: {e}!')

mindsdb/integrations/handlers/google_books_handler/google_books_tables.py

@@ -3,9 +3,6 @@ from mindsdb_sql_parser import ast
 from pandas import DataFrame
 
 from mindsdb.integrations.libs.api_handler import APITable
-from mindsdb.integrations.libs.response import (
-    HandlerResponse as Response,
-)
 from mindsdb.integrations.utilities.sql_utils import extract_comparison_conditions
 
 

mindsdb/integrations/handlers/google_calendar_handler/google_calendar_handler.py

@@ -9,7 +9,7 @@ from mindsdb.integrations.libs.response import (
 )
 from mindsdb.utilities.config import Config
 from mindsdb.utilities import log
-from mindsdb.integrations.utilities.handlers.auth_utilities import GoogleUserOAuth2Manager
+from mindsdb.integrations.utilities.handlers.auth_utilities.google import GoogleUserOAuth2Manager
 from mindsdb.integrations.utilities.handlers.auth_utilities.exceptions import AuthException
 
 from .google_calendar_tables import GoogleCalendarEventsTable
@@ -177,8 +177,8 @@ class GoogleCalendarHandler(APIHandler):
             'recurrence': [
                 'RRULE:FREQ=DAILY;COUNT=1'
             ],
-            'attendees': [{'email': attendee['email']} for attendee in (params['attendees'] 
-                            if isinstance(params['attendees'], list) else [params['attendees']])],
+            'attendees': [{'email': attendee['email']} for attendee in (params['attendees']
+                                                                        if isinstance(params['attendees'], list) else [params['attendees']])],
             'reminders': {
                 'useDefault': False,
                 'overrides': [
@@ -188,7 +188,7 @@ class GoogleCalendarHandler(APIHandler):
             },
         }
 
-        event = service.events().insert(calendarId='primary', 
+        event = service.events().insert(calendarId='primary',
                                         body=event).execute()
         return pd.DataFrame([event], columns=self.events.get_columns())
 

mindsdb/integrations/handlers/google_calendar_handler/google_calendar_tables.py

@@ -4,9 +4,6 @@ from mindsdb_sql_parser import ast
 from pandas import DataFrame
 
 from mindsdb.integrations.libs.api_handler import APITable
-from mindsdb.integrations.libs.response import (
-    HandlerResponse as Response,
-)
 from mindsdb.integrations.utilities.date_utils import utc_date_str_to_timestamp_ms, parse_utc_date
 from mindsdb.integrations.utilities.sql_utils import extract_comparison_conditions
 
@@ -41,7 +38,6 @@ class GoogleCalendarEventsTable(APITable):
                 params[arg1] = arg2
             elif arg1 == 'q':
                 params[arg1] = arg2
-           
 
         # Get the order by from the query.
         if query.order_by is not None:
@@ -93,7 +89,7 @@ class GoogleCalendarEventsTable(APITable):
         event_data = {}
         timestamp_columns = {'start_time', 'end_time', 'created', 'updated'}
         regular_columns = {'summary', 'description', 'location', 'status', 'html_link',
-                            'creator', 'organizer', 'reminders', 'timeZone', 'calendar_id', 'attendees'}
+                           'creator', 'organizer', 'reminders', 'timeZone', 'calendar_id', 'attendees'}
 
         # TODO: check why query.columns is None
         for col, val in zip(query.columns, values):
@@ -103,7 +99,7 @@ class GoogleCalendarEventsTable(APITable):
                 event_data[col.name] = val
             else:
                 raise NotImplementedError
-            
+
         st = datetime.datetime.utcfromtimestamp(event_data['start_time'] / 1000).isoformat() + 'Z'
         et = datetime.datetime.utcfromtimestamp(event_data['end_time'] / 1000).isoformat() + 'Z'
 

mindsdb/integrations/handlers/google_content_shopping_handler/google_content_shopping_handler.py

@@ -17,6 +17,7 @@ from mindsdb.utilities import log
 
 logger = log.getLogger(__name__)
 
+
 class GoogleContentShoppingHandler(APIHandler):
     """
         A class for handling connections and interactions with the Google Content API for Shopping.
@@ -88,7 +89,7 @@ class GoogleContentShoppingHandler(APIHandler):
         response = StatusResponse(False)
 
         try:
-            service = self.connect()
+            self.connect()
             response.success = True
         except Exception as e:
             logger.error(f'Error connecting to Google Content API for Shopping: {e}!')
@@ -205,7 +206,7 @@ class GoogleContentShoppingHandler(APIHandler):
                                                                    'placedDateStart',
                                                                    'placedDateEnd',
                                                                    'orderBy']
-                                                           and value is not None
+            and value is not None
         }
         if params['order_id']:
             result = service.orders().get(merchantId=self.merchant_id, orderId=params['order_id'], **args).execute()

mindsdb/integrations/handlers/google_content_shopping_handler/google_content_shopping_tables.py

@@ -3,9 +3,6 @@ from mindsdb_sql_parser import ast
 from pandas import DataFrame
 
 from mindsdb.integrations.libs.api_handler import APITable
-from mindsdb.integrations.libs.response import (
-    HandlerResponse as Response,
-)
 from mindsdb.integrations.utilities.date_utils import parse_utc_date
 from mindsdb.integrations.utilities.sql_utils import extract_comparison_conditions
 

mindsdb/integrations/handlers/google_fit_handler/google_fit_handler.py

@@ -15,7 +15,7 @@ from mindsdb_sql_parser import parse_sql
 
 from mindsdb.utilities import log
 from mindsdb.integrations.handlers.google_fit_handler.google_fit_tables import GoogleFitTable
-from mindsdb.integrations.libs.api_handler import APIHandler,FuncParser
+from mindsdb.integrations.libs.api_handler import APIHandler
 from mindsdb.integrations.libs.response import (
     HandlerStatusResponse as StatusResponse,
     HandlerResponse as Response,
@@ -46,7 +46,6 @@ class GoogleFitHandler(APIHandler):
             self.credentials_path = 'mindsdb/integrations/handlers/google_fit_handler/credentials.json'
         else:
             raise Exception('Connection args have to content ether service_account_file or service_account_json')
-        
 
         self.api = None
         self.is_connected = False
@@ -58,11 +57,11 @@ class GoogleFitHandler(APIHandler):
         if self.is_connected is True and self.api:
             return self.api
         if self.connection_args:
-            credentialDict = {"installed":self.connection_args}
+            credentialDict = {"installed": self.connection_args}
             f = open(self.credentials_path, "w")
             f.write(json.dumps(credentialDict).replace(" ", ""))
             f.close()
-            
+
         creds = None
 
         if os.path.isfile('mindsdb/integrations/handlers/google_fit_handler/token.json'):
@@ -77,7 +76,7 @@ class GoogleFitHandler(APIHandler):
             with open('mindsdb/integrations/handlers/google_fit_handler/token.json', 'w') as token:
                 token.write(creds.to_json())
         self.api = build('fitness', 'v1', credentials=creds)
-        
+
         self.is_connected = True
         return self.api
 
@@ -85,7 +84,7 @@ class GoogleFitHandler(APIHandler):
         response = StatusResponse(False)
 
         try:
-            api = self.connect()
+            self.connect()
             response.success = True
 
         except Exception as e:
@@ -109,7 +108,6 @@ class GoogleFitHandler(APIHandler):
         except HttpError:
             raise HttpError
 
-
     def native_query(self, query: str = None) -> Response:
         """Receive raw query and act upon it somehow.
         Args:
@@ -120,13 +118,13 @@ class GoogleFitHandler(APIHandler):
         """
         ast = parse_sql(query)
         return self.query(ast)
-    
+
     def get_steps(self, start_time_millis, end_time_millis) -> pd.DataFrame:
         steps = {}
         steps_data = self.retrieve_data(self.api, start_time_millis, end_time_millis, "derived:com.google.step_count.delta:com.google.android.gms:estimated_steps")
         for daily_step_data in steps_data['bucket']:
             local_date = datetime.fromtimestamp(int(daily_step_data['startTimeMillis']) / 1000,
-                                            tz=pytz.timezone(str(get_localzone())))
+                                                tz=pytz.timezone(str(get_localzone())))
             local_date_str = local_date.strftime(DATE_FORMAT)
 
             data_point = daily_step_data['dataset'][0]['point']
@@ -139,8 +137,8 @@ class GoogleFitHandler(APIHandler):
         ret = ret.drop('originDataSourceId', axis=1)
         ret = ret.reset_index(drop=False)
         return ret
-    
-    def call_google_fit_api(self, method_name:str = None, params:dict = None) -> pd.DataFrame:
+
+    def call_google_fit_api(self, method_name: str = None, params: dict = None) -> pd.DataFrame:
         """Receive query as AST (abstract syntax tree) and act upon it somehow.
         Args:
             query (ASTNode): sql query represented as AST. May be any kind
@@ -152,4 +150,4 @@ class GoogleFitHandler(APIHandler):
         if method_name == 'get_steps':
             val = self.get_steps(params['start_time'], params['end_time'])
             return val
-        raise NotImplementedError('Method name {} not supported by Google Fit Handler'.format(method_name))
+        raise NotImplementedError('Method name {} not supported by Google Fit Handler'.format(method_name))

mindsdb/integrations/handlers/google_fit_handler/google_fit_tables.py

@@ -7,6 +7,7 @@ import pytz
 import time
 from tzlocal import get_localzone
 
+
 class GoogleFitTable(APITable):
 
     def time_parser(self, args) -> int:
@@ -21,38 +22,35 @@ class GoogleFitTable(APITable):
         epoch0 = datetime.datetime(1970, 1, 1, tzinfo=pytz.utc)
         time = pytz.timezone(str(get_localzone())).localize(datetime.datetime(int(ymd[0].rstrip()), int(ymd[1].rstrip()), int(ymd[2].rstrip())))
         return int((time - epoch0).total_seconds() * 1000)
-    
+
     def select(self, query: ast.Select) -> Response:
 
         conditions = extract_comparison_conditions(query.where)
-        
+
         params = {}
-        filters = []
-        steps = {}
-        #get the local time
+        # get the local time
         now = int(round(time.time() * 1000))
 
-        #hard coded for now as user default query time period
-        one_year = 31536000000
+        # hard coded for now as user default query time period
         one_month = 2629746000
         for op, arg1, arg2 in conditions:
             if op == 'or':
-                raise NotImplementedError(f'OR is not supported')
+                raise NotImplementedError('OR is not supported')
             if arg1 == 'date':
                 date = self.time_parser(arg2)
                 if op == '>':
                     params['start_time'] = date
                     params['end_time'] = now
-                
-                #hard coded as a month
+
+                # hard coded as a month
                 elif op == '<':
                     params['start_time'] = date - one_month
                     params['end_time'] = date
                 else:
                     raise NotImplementedError
             else:
-                raise NotImplementedError(f'This query is not supported')
-        # if time is not provided in the query, the time range is one month ago to now 
+                raise NotImplementedError('This query is not supported')
+        # if time is not provided in the query, the time range is one month ago to now
         if not params:
             params['start_time'] = now - one_month
             params['end_time'] = now
@@ -66,4 +64,4 @@ class GoogleFitTable(APITable):
         return [
             'dates',
             'steps'
-        ]     
+        ]

mindsdb/integrations/handlers/google_search_handler/google_search_handler.py

@@ -16,6 +16,7 @@ from mindsdb.utilities import log
 
 logger = log.getLogger(__name__)
 
+
 class GoogleSearchConsoleHandler(APIHandler):
     """
         A class for handling connections and interactions with the Google Search Console API.
@@ -86,7 +87,7 @@ class GoogleSearchConsoleHandler(APIHandler):
         response = StatusResponse(False)
 
         try:
-            service = self.connect()
+            self.connect()
             response.success = True
         except Exception as e:
             logger.error(f'Error connecting to Google Search Console API: {e}!')

mindsdb/integrations/handlers/google_search_handler/google_search_tables.py

@@ -3,9 +3,6 @@ from mindsdb_sql_parser import ast
 from pandas import DataFrame
 
 from mindsdb.integrations.libs.api_handler import APITable
-from mindsdb.integrations.libs.response import (
-    HandlerResponse as Response,
-)
 from mindsdb.integrations.utilities.date_utils import parse_utc_date
 from mindsdb.integrations.utilities.sql_utils import extract_comparison_conditions
 

mindsdb/integrations/handlers/groq_handler/__init__.py

@@ -1,10 +1,10 @@
+from .__about__ import __version__ as version
+from .__about__ import __description__ as description
 from mindsdb.integrations.libs.const import HANDLER_TYPE
 from mindsdb.utilities import log
 
 logger = log.getLogger(__name__)
 
-from .__about__ import __description__ as description
-from .__about__ import __version__ as version
 
 try:
     from .groq_handler import GroqHandler as Handler
@@ -20,4 +20,4 @@ type = HANDLER_TYPE.ML
 icon_path = 'icon.svg'
 permanent = False
 
-__all__ = ["Handler", "version", "name", "type", "title", "description", "import_error", "icon_path"]
+__all__ = ["Handler", "version", "name", "type", "title", "description", "import_error", "icon_path"]

mindsdb/integrations/handlers/hackernews_handler/hn_handler.py

@@ -1,14 +1,13 @@
 import requests
 import pandas as pd
 from mindsdb.utilities import log
-from mindsdb.utilities.config import Config
-from mindsdb_sql_parser import ast
-from mindsdb.integrations.libs.api_handler import APIHandler, APITable
+from mindsdb.integrations.libs.api_handler import APIHandler
 from mindsdb.integrations.libs.response import HandlerStatusResponse as StatusResponse, HandlerResponse as Response, RESPONSE_TYPE
-from .hn_table import StoriesTable, CommentsTable , HNStoriesTable ,JobStoriesTable, ShowStoriesTable
+from .hn_table import StoriesTable, CommentsTable, HNStoriesTable, JobStoriesTable, ShowStoriesTable
 
 logger = log.getLogger(__name__)
 
+
 class HackerNewsHandler(APIHandler):
     """
     A class for handling connections and interactions with the Hacker News API.
@@ -23,10 +22,10 @@ class HackerNewsHandler(APIHandler):
         self._register_table('stories', stories)
 
         hnstories = HNStoriesTable(self)
-        self._register_table('hnstories',hnstories)
+        self._register_table('hnstories', hnstories)
 
         jobstories = JobStoriesTable(self)
-        self._register_table('jobstories',jobstories)
+        self._register_table('jobstories', jobstories)
 
         showstories = ShowStoriesTable(self)
         self._register_table('showstories', showstories)
@@ -99,4 +98,3 @@ class HackerNewsHandler(APIHandler):
             raise ValueError(f'Unknown method_name: {method_name}')
 
         return df
-

mindsdb/integrations/handlers/hackernews_handler/hn_table.py

@@ -2,8 +2,7 @@ import pandas as pd
 from mindsdb.integrations.libs.api_handler import APITable
 from mindsdb_sql_parser import ast
 from mindsdb.integrations.utilities.sql_utils import extract_comparison_conditions
-from typing import List, Tuple
-import requests
+from typing import List
 
 
 class StoriesTable(APITable):
@@ -40,7 +39,6 @@ class StoriesTable(APITable):
 
         return df
 
-
     def get_columns(self):
         """Get the list of column names for the stories table.
         Returns:
@@ -63,15 +61,17 @@ class StoriesTable(APITable):
                 columns.append(target.value)
         df = df[columns]
         return df
-    
+
+
 class HNStoriesTable(StoriesTable):
     json_endpoint = "askstories.json"
     columns = ['id', 'time', 'title', 'text', 'score', 'descendants']
 
+
 class JobStoriesTable(StoriesTable):
     json_endpoint = "jobstories.json"
     columns = ['id', 'time', 'title', 'url', 'score', 'type']
-    
+
 
 class ShowStoriesTable(StoriesTable):
     json_endpoint = "showstories.json"
@@ -116,8 +116,7 @@ class CommentsTable(APITable):
             comments_df = comments_df.head(limit)
 
         return comments_df
-    
-    
+
     def get_columns(self) -> List[str]:
         """Get the list of column names for the comments table.
         Returns:

mindsdb/integrations/handlers/hive_handler/tests/test_hive_handler.py

@@ -47,4 +47,4 @@ class HiveHandlerTest(unittest.TestCase):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()