ruff

python/pyproject.toml

@@ -57,7 +57,7 @@ tests = [
     "duckdb",
     "pytz",
     "polars>=0.19",
-    "PIL"
+    "pillow",
 ]
 dev = ["ruff", "pre-commit"]
 docs = [

python/python/lancedb/pydantic.py

@@ -36,7 +36,6 @@ from typing import (
 import numpy as np
 import pyarrow as pa
 import pydantic
-from .util import attempt_import_or_raise
 import semver
 from lance.arrow import (
     EncodedImageType,
@@ -44,6 +43,8 @@ from lance.arrow import (
 from pydantic.fields import FieldInfo
 from pydantic_core import core_schema
 
+from .util import attempt_import_or_raise
+
 PYDANTIC_VERSION = semver.Version.parse(pydantic.__version__)
 
 if TYPE_CHECKING:
@@ -204,7 +205,7 @@ class ImageMixin(ABC):
 
 
 def EncodedImage():
-    attempt_import_or_raise("PIL", "PIL or pip install lancedb[embeddings]")
+    attempt_import_or_raise("PIL", "pillow or pip install lancedb[embeddings]")
     import PIL.Image
 
     class EncodedImage(bytes, ImageMixin):

python/python/tests/test_cli.py

@@ -1,5 +1,4 @@
 from click.testing import CliRunner
-
 from lancedb.cli.cli import cli
 from lancedb.utils import CONFIG
 

python/python/tests/test_context.py

@@ -13,7 +13,6 @@
 
 import pandas as pd
 import pytest
-
 from lancedb.context import contextualize
 
 

python/python/tests/test_db.py

@@ -19,8 +19,6 @@ import numpy as np
 import pandas as pd
 import pyarrow as pa
 import pytest
-
-import lancedb
 from lancedb.pydantic import LanceModel, Vector
 
 

python/python/tests/test_e2e_remote_db.py

@@ -13,7 +13,6 @@
 
 import numpy as np
 import pytest
-
 from lancedb import LanceDBConnection
 
 # TODO: setup integ test mark and script

python/python/tests/test_embeddings.py

@@ -13,11 +13,10 @@
 import sys
 
 import lance
+import lancedb
 import numpy as np
 import pyarrow as pa
 import pytest
-
-import lancedb
 from lancedb.conftest import MockTextEmbeddingFunction
 from lancedb.embeddings import (
     EmbeddingFunctionConfig,

python/python/tests/test_embeddings_slow.py

@@ -14,12 +14,11 @@ import importlib
 import io
 import os
 
+import lancedb
 import numpy as np
 import pandas as pd
 import pytest
 import requests
-
-import lancedb
 from lancedb.embeddings import get_registry
 from lancedb.pydantic import LanceModel, Vector
 
@@ -185,10 +184,9 @@ def test_imagebind(tmp_path):
     import shutil
     import tempfile
 
+    import lancedb.embeddings.imagebind
     import pandas as pd
     import requests
-
-    import lancedb.embeddings.imagebind
     from lancedb.embeddings import get_registry
     from lancedb.pydantic import LanceModel, Vector
 

python/python/tests/test_fts.py

@@ -14,12 +14,11 @@ import os
 import random
 from unittest import mock
 
+import lancedb as ldb
 import numpy as np
 import pandas as pd
 import pytest
 
-import lancedb as ldb
-
 pytest.importorskip("lancedb.fts")
 tantivy = pytest.importorskip("tantivy")
 

python/python/tests/test_io.py

@@ -13,9 +13,8 @@
 
 import os
 
-import pytest
-
 import lancedb
+import pytest
 
 # You need to setup AWS credentials an a base path to run this test. Example
 #    AWS_PROFILE=default TEST_S3_BASE_URL=s3://my_bucket/dataset pytest tests/test_io.py

python/python/tests/test_query.py

@@ -18,7 +18,6 @@ import numpy as np
 import pandas.testing as tm
 import pyarrow as pa
 import pytest
-
 from lancedb.db import LanceDBConnection
 from lancedb.pydantic import LanceModel, Vector
 from lancedb.query import LanceVectorQueryBuilder, Query

python/python/tests/test_remote_client.py

@@ -17,7 +17,6 @@ import pandas as pd
 import pyarrow as pa
 import pytest
 from aiohttp import web
-
 from lancedb.remote.client import RestfulLanceDBClient, VectorQuery
 
 

python/python/tests/test_remote_db.py

@@ -11,9 +11,8 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
-import pyarrow as pa
-
 import lancedb
+import pyarrow as pa
 from lancedb.remote.client import VectorQuery, VectorQueryResult
 
 

python/python/tests/test_rerankers.py

@@ -1,9 +1,8 @@
 import os
 
+import lancedb
 import numpy as np
 import pytest
-
-import lancedb
 from lancedb.conftest import MockTextEmbeddingFunction  # noqa
 from lancedb.embeddings import EmbeddingFunctionRegistry
 from lancedb.pydantic import LanceModel, Vector

python/python/tests/test_telemetry.py

@@ -1,8 +1,7 @@
 import json
 
-import pytest
-
 import lancedb
+import pytest
 from lancedb.utils.events import _Events
 
 

python/python/tests/test_util.py

@@ -15,7 +15,6 @@ import os
 import pathlib
 
 import pytest
-
 from lancedb.util import get_uri_scheme, join_uri