linting

logan-markewich · Nov 22, 2023 · c8861c8 · c8861c8
1 parent 321cdf7
commit c8861c8
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 11 deletions.
diff --git a/llama_hub/llama_packs/retry_engine_weaviate/base.py b/llama_hub/llama_packs/retry_engine_weaviate/base.py
@@ -5,18 +5,11 @@
 
 from llama_index.indices.vector_store import VectorStoreIndex
 from llama_index.llama_pack.base import BaseLlamaPack
-from llama_index.query_engine import RetrieverQueryEngine
 from llama_index.schema import TextNode
 from llama_index.storage.storage_context import StorageContext
 from llama_index.vector_stores.weaviate import WeaviateVectorStore
 from llama_index.vector_stores.types import VectorStoreInfo
-from llama_index.tools import QueryEngineTool, ToolMetadata
-from llama_index.query_engine import SubQuestionQueryEngine
-from llama_index.evaluation.guideline_eval import GuidelineEvaluator, DEFAULT_GUIDELINES
-from llama_index.response.schema import Response
-from llama_index.indices.query.query_transform.feedback_transform import (
-    FeedbackQueryTransformation,
-)
+from llama_index.evaluation.guideline import GuidelineEvaluator, DEFAULT_GUIDELINES
 from llama_index.query_engine.retry_query_engine import (
     RetryGuidelineQueryEngine,
 )

diff --git a/llama_hub/llama_packs/sub_question_weaviate/base.py b/llama_hub/llama_packs/sub_question_weaviate/base.py
@@ -5,7 +5,6 @@
 
 from llama_index.indices.vector_store import VectorStoreIndex
 from llama_index.llama_pack.base import BaseLlamaPack
-from llama_index.query_engine import RetrieverQueryEngine
 from llama_index.schema import TextNode
 from llama_index.storage.storage_context import StorageContext
 from llama_index.vector_stores.weaviate import WeaviateVectorStore

diff --git a/llama_hub/llama_packs/timescale_vector_autoretrieval/__init__.py b/llama_hub/llama_packs/timescale_vector_autoretrieval/__init__.py
@@ -1,3 +1,5 @@
-from llama_hub.llama_packs.timescale_vector_autoretrieval.base import TimescaleVectorAutoretrievalPack
+from llama_hub.llama_packs.timescale_vector_autoretrieval.base import (
+    TimescaleVectorAutoretrievalPack,
+)
 
 __all__ = ["TimescaleVectorAutoretrievalPack"]
diff --git a/llama_hub/llama_packs/timescale_vector_autoretrieval/base.py b/llama_hub/llama_packs/timescale_vector_autoretrieval/base.py
@@ -34,7 +34,8 @@ def __init__(
         self._vector_store = TimescaleVectorStore.from_params(
             service_url=service_url,
             table_name=table_name,
-            time_partition_interval=time_partition_interval)
+            time_partition_interval=time_partition_interval,
+        )
 
         if nodes is not None:
             self._storage_context = StorageContext.from_defaults(