isort entire repo (#200)

python/lancedb/context.py

@@ -13,7 +13,8 @@
 from __future__ import annotations
 
 import pandas as pd
-from .exceptions import MissingValueError, MissingColumnError
+
+from .exceptions import MissingColumnError, MissingValueError
 
 
 def contextualize(raw_df: pd.DataFrame) -> Contextualizer:

python/lancedb/db.py

@@ -12,18 +12,17 @@
 #  limitations under the License.
 
 from __future__ import annotations
-import functools
 
+import functools
 import os
 from pathlib import Path
-import os
 
 import pyarrow as pa
 from pyarrow import fs
 
 from .common import DATA, URI
 from .table import LanceTable
-from .util import get_uri_scheme, get_uri_location
+from .util import get_uri_location, get_uri_scheme
 
 
 class LanceDBConnection:

python/lancedb/query.py

@@ -11,12 +11,13 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 from __future__ import annotations
+
+import asyncio
 from typing import Awaitable, Literal
 
 import numpy as np
 import pandas as pd
 import pyarrow as pa
-import asyncio
 
 from .common import VECTOR_COLUMN_NAME
 

python/lancedb/remote/__init__.py

@@ -13,10 +13,10 @@
 
 import abc
 from typing import List, Optional
+
 import attr
 import pandas as pd
 import pyarrow as pa
-
 from pydantic import BaseModel
 
 __all__ = ["LanceDBClient", "VectorQuery", "VectorQueryResult"]

python/lancedb/remote/client.py

@@ -13,11 +13,11 @@
 
 
 import functools
+import urllib.parse
 
 import aiohttp
 import attr
 import pyarrow as pa
-import urllib.parse
 
 from lancedb.remote import VectorQuery, VectorQueryResult
 from lancedb.remote.errors import LanceDBClientError

python/tests/test_e2e_remote_db.py

@@ -11,12 +11,11 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
-from lancedb import LanceDBConnection
-
 import numpy as np
-
 import pytest
 
+from lancedb import LanceDBConnection
+
 # TODO: setup integ test mark and script
 
 

python/tests/test_embeddings.py

@@ -14,6 +14,7 @@ import sys
 
 import numpy as np
 import pyarrow as pa
+
 from lancedb.embeddings import with_embeddings
 
 

python/tests/test_fts.py

@@ -13,13 +13,13 @@
 import os
 import random
 
-import lancedb.fts
 import numpy as np
 import pandas as pd
 import pytest
 import tantivy
 
 import lancedb as ldb
+import lancedb.fts
 
 
 @pytest.fixture

python/tests/test_io.py

@@ -12,6 +12,7 @@
 #  limitations under the License.
 
 import os
+
 import pytest
 
 import lancedb

python/tests/test_query.py

@@ -17,9 +17,9 @@ import pandas as pd
 import pandas.testing as tm
 import pyarrow as pa
 import pytest
-from lancedb.query import LanceQueryBuilder
 
 from lancedb.db import LanceDBConnection
+from lancedb.query import LanceQueryBuilder
 
 
 class MockTable:

python/tests/test_remote_db.py

@@ -13,9 +13,8 @@
 
 import pyarrow as pa
 
-
-from lancedb.remote.client import VectorQuery, VectorQueryResult
 from lancedb.db import LanceDBConnection
+from lancedb.remote.client import VectorQuery, VectorQueryResult
 
 
 class FakeLanceDBClient:

python/tests/test_table.py

@@ -17,6 +17,7 @@ from pathlib import Path
 import pandas as pd
 import pyarrow as pa
 import pytest
+
 from lancedb.table import LanceTable