Linter Fix

googleapis · Nov 27, 2024 · f9df393 · f9df393
1 parent 11fb1ba
commit f9df393
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/src/llama_index_alloydb_pg/async_index_store.py b/src/llama_index_alloydb_pg/async_index_store.py
@@ -15,8 +15,8 @@
 from __future__ import annotations
 
 import json
-from typing import List, Optional
 import warnings
+from typing import List, Optional
 
 from llama_index.core.constants import DATA_KEY
 from llama_index.core.data_structs.data_structs import IndexStruct

diff --git a/tests/test_async_index_store.py b/tests/test_async_index_store.py
@@ -14,8 +14,8 @@
 
 import os
 import uuid
-from typing import Sequence
 import warnings
+from typing import Sequence
 
 import pytest
 import pytest_asyncio
@@ -163,7 +163,9 @@ async def test_warning(self, index_store):
         await index_store.aadd_index_struct(index_list_struct)
 
         with warnings.catch_warnings(record=True) as w:
-          index_struct = await index_store.aget_index_struct()
+            index_struct = await index_store.aget_index_struct()
 
-          assert len(w) == 1
-          assert "No struct_id specified and more than one struct exists." in str(w[-1].message)
+            assert len(w) == 1
+            assert "No struct_id specified and more than one struct exists." in str(
+                w[-1].message
+            )