From fec3e158b5dd596e66e7271cc8f8933eca4704f8 Mon Sep 17 00:00:00 2001 From: yulaicui Date: Mon, 10 Jul 2023 13:59:49 -0400 Subject: [PATCH] linter --- apps/youtube_qa/youtube_qa.py | 4 +++- evadb/interfaces/relational/db.py | 11 +++++++++-- evadb/parser/utils.py | 7 +++++-- test/interfaces/relational/test_relational_api.py | 4 +++- 4 files changed, 20 insertions(+), 6 deletions(-) diff --git a/apps/youtube_qa/youtube_qa.py b/apps/youtube_qa/youtube_qa.py index 8bfad33f6b..5687be1cef 100644 --- a/apps/youtube_qa/youtube_qa.py +++ b/apps/youtube_qa/youtube_qa.py @@ -216,7 +216,9 @@ def generate_online_video_transcript(cursor: evadb.EvaDBCursor) -> str: # extract speech texts from videos cursor.drop_table("youtube_video_text", if_exists=True).execute() - speech_recognizer_rel = cursor.table("youtube_video").select("SpeechRecognizer(audio)") + speech_recognizer_rel = cursor.table("youtube_video").select( + "SpeechRecognizer(audio)" + ) cursor.create_table("youtube_video_text", query=speech_recognizer_rel).execute() print("✅ Video analysis completed.") diff --git a/evadb/interfaces/relational/db.py b/evadb/interfaces/relational/db.py index c1396e516d..56f5e11373 100644 --- a/evadb/interfaces/relational/db.py +++ b/evadb/interfaces/relational/db.py @@ -391,7 +391,12 @@ def create_function( return EvaDBQuery(self._evadb, stmt) def create_table( - self, table_name: str, if_not_exists: bool = True, columns: str = None, query: EvaDBQuery = None, **kwargs + self, + table_name: str, + if_not_exists: bool = True, + columns: str = None, + query: EvaDBQuery = None, + **kwargs, ) -> "EvaDBQuery": """ Create a udf in the database. @@ -424,7 +429,9 @@ def create_table( stmt = parse_create_table(table_name, if_not_exists, columns, **kwargs) else: select_query = query.sql_query() - stmt = parse_create_table_from_select(table_name, if_not_exists, select_query, **kwargs) + stmt = parse_create_table_from_select( + table_name, if_not_exists, select_query, **kwargs + ) return EvaDBQuery(self._evadb, stmt) def query(self, sql_query: str) -> EvaDBQuery: diff --git a/evadb/parser/utils.py b/evadb/parser/utils.py index 5c42a6aae4..eb9375ecf0 100644 --- a/evadb/parser/utils.py +++ b/evadb/parser/utils.py @@ -85,7 +85,10 @@ def parse_create_table(table_name: str, if_not_exists: bool, columns: str, **kwa assert isinstance(stmt, CreateTableStatement), "Expected a create table statement" return stmt -def parse_create_table_from_select(table_name: str, if_not_exists: bool, query: str, **kwargs): + +def parse_create_table_from_select( + table_name: str, if_not_exists: bool, query: str, **kwargs +): mock_query = ( f"CREATE TABLE IF NOT EXISTS {table_name} AS {query};" if if_not_exists @@ -93,7 +96,7 @@ def parse_create_table_from_select(table_name: str, if_not_exists: bool, query: ) stmt = Parser().parse(mock_query)[0] assert isinstance(stmt, CreateTableStatement), "Expected a create table statement" - return stmt + return stmt def parse_show(show_type: str, **kwargs): diff --git a/test/interfaces/relational/test_relational_api.py b/test/interfaces/relational/test_relational_api.py index 3b9b1e013e..9e908ecfe1 100644 --- a/test/interfaces/relational/test_relational_api.py +++ b/test/interfaces/relational/test_relational_api.py @@ -493,7 +493,9 @@ def test_create_table_from_select_relational(self): create_dummy_object_detector_udf.execute() # Check create table from select relation - select_query_sql_rel = cursor.table("dummy_video").select("id, DummyObjectDetector(data)") + select_query_sql_rel = cursor.table("dummy_video").select( + "id, DummyObjectDetector(data)" + ) cursor.drop_table("dummy_objects", if_exists=True) cursor.create_table("dummy_objects", query=select_query_sql_rel).execute()