From 62e3a2284e99144cd052afaa9afb59dae133d756 Mon Sep 17 00:00:00 2001 From: Mohammad Amin Date: Mon, 20 May 2024 12:35:11 +0330 Subject: [PATCH] fix: linter issues! --- tests/unit/test_base_pg_engine.py | 2 +- tests/unit/test_gdrive_query_engine.py | 4 +++- tests/unit/test_github_query_engine.py | 4 +++- tests/unit/test_mediawiki_query_engine.py | 4 +++- tests/unit/test_notion_query_engine.py | 4 +++- utils/query_engine/base_pg_engine.py | 2 +- 6 files changed, 14 insertions(+), 6 deletions(-) diff --git a/tests/unit/test_base_pg_engine.py b/tests/unit/test_base_pg_engine.py index 15c2c7b..1e2c815 100644 --- a/tests/unit/test_base_pg_engine.py +++ b/tests/unit/test_base_pg_engine.py @@ -21,4 +21,4 @@ def test_setup_vector_store_index(self): expected_dbname = f"community_{community_id}" self.assertIn(expected_dbname, base_engine.vector_store.connection_string) - self.assertEqual(base_engine.vector_store.table_name, platform_table_name) \ No newline at end of file + self.assertEqual(base_engine.vector_store.table_name, platform_table_name) diff --git a/tests/unit/test_gdrive_query_engine.py b/tests/unit/test_gdrive_query_engine.py index 7ab4e65..3aac35b 100644 --- a/tests/unit/test_gdrive_query_engine.py +++ b/tests/unit/test_gdrive_query_engine.py @@ -1,6 +1,8 @@ from unittest import TestCase -from llama_index.core.indices.vector_store.retrievers.retriever import VectorIndexRetriever +from llama_index.core.indices.vector_store.retrievers.retriever import ( + VectorIndexRetriever, +) from utils.query_engine import GDriveQueryEngine diff --git a/tests/unit/test_github_query_engine.py b/tests/unit/test_github_query_engine.py index b3b657f..3d77abb 100644 --- a/tests/unit/test_github_query_engine.py +++ b/tests/unit/test_github_query_engine.py @@ -1,6 +1,8 @@ from unittest import TestCase -from llama_index.core.indices.vector_store.retrievers.retriever import VectorIndexRetriever +from llama_index.core.indices.vector_store.retrievers.retriever import ( + VectorIndexRetriever, +) from utils.query_engine import GitHubQueryEngine diff --git a/tests/unit/test_mediawiki_query_engine.py b/tests/unit/test_mediawiki_query_engine.py index 303aa63..235a0ae 100644 --- a/tests/unit/test_mediawiki_query_engine.py +++ b/tests/unit/test_mediawiki_query_engine.py @@ -1,6 +1,8 @@ from unittest import TestCase -from llama_index.core.indices.vector_store.retrievers.retriever import VectorIndexRetriever +from llama_index.core.indices.vector_store.retrievers.retriever import ( + VectorIndexRetriever, +) from utils.query_engine import MediaWikiQueryEngine diff --git a/tests/unit/test_notion_query_engine.py b/tests/unit/test_notion_query_engine.py index 57f2b96..05587a6 100644 --- a/tests/unit/test_notion_query_engine.py +++ b/tests/unit/test_notion_query_engine.py @@ -1,6 +1,8 @@ from unittest import TestCase -from llama_index.core.indices.vector_store.retrievers.retriever import VectorIndexRetriever +from llama_index.core.indices.vector_store.retrievers.retriever import ( + VectorIndexRetriever, +) from utils.query_engine import NotionQueryEngine diff --git a/utils/query_engine/base_pg_engine.py b/utils/query_engine/base_pg_engine.py index 1f2e415..7b2d16a 100644 --- a/utils/query_engine/base_pg_engine.py +++ b/utils/query_engine/base_pg_engine.py @@ -77,4 +77,4 @@ def _setup_vector_store_index( embed_model=embed_model, ) index = pg_vector.load_index() - return index \ No newline at end of file + return index