fix: linter issues!

TogetherCrew · May 20, 2024 · 62e3a22 · 62e3a22
1 parent 04581b9
commit 62e3a22
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 6 deletions.
diff --git 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)
+        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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -77,4 +77,4 @@ def _setup_vector_store_index(
             embed_model=embed_model,
         )
         index = pg_vector.load_index()
-        return index
+        return index