Gaetan Lepage
2024-09-13 20:33:44 +02:00
parent 6b193b59ec
commit ccd52f9881
3 changed files with 316 additions and 294 deletions
File diff suppressed because it is too large Load Diff
@@ -28,19 +28,16 @@
buildPythonPackage rec {
pname = "lancedb";
version = "0.12.0";
version = "0.13.0";
pyproject = true;
src = fetchFromGitHub {
owner = "lancedb";
repo = "lancedb";
rev = "refs/tags/python-v${version}";
hash = "sha256-LDxq49aFxUmRAw8tQvFxnExtU0IKKqMbxIBuY95cBHU=";
hash = "sha256-6E20WgyoEALdxmiOfgq89dCkqovvIMzc/wy+kvjDWwU=";
};
# ratelimiter only support up to python310 and it has been removed from nixpkgs
patches = [ ./remove-ratelimiter.patch ];
buildAndTestSubdir = "python";
cargoDeps = rustPlatform.importCargoLock { lockFile = ./Cargo.lock; };
@@ -70,8 +67,6 @@ buildPythonPackage rec {
]
);
pythonRemoveDeps = [ "ratelimiter" ];
dependencies = [
attrs
cachetools
@@ -104,6 +99,10 @@ buildPythonPackage rec {
disabledTests = [
# require tantivy which is not packaged in nixpkgs
"test_basic"
# polars.exceptions.ComputeError: TypeError: _scan_pyarrow_dataset_impl() got multiple values for argument 'batch_size'
# https://github.com/lancedb/lancedb/issues/1539
"test_polars"
];
disabledTestPaths = [
@@ -1,26 +0,0 @@
diff --git a/python/python/lancedb/embeddings/utils.py b/python/python/lancedb/embeddings/utils.py
index 813631c..4ae6025 100644
--- a/python/python/lancedb/embeddings/utils.py
+++ b/python/python/lancedb/embeddings/utils.py
@@ -111,7 +111,7 @@ class FunctionWrapper:
if len(self.rate_limiter_kwargs) > 0:
v = int(sys.version_info.minor)
- if v >= 11:
+ if True:
print(
"WARNING: rate limit only support up to 3.10, proceeding "
"without rate limiter"
diff --git a/python/python/tests/test_embeddings.py b/python/python/tests/test_embeddings.py
index ed7b105..28ca9cb 100644
--- a/python/python/tests/test_embeddings.py
+++ b/python/python/tests/test_embeddings.py
@@ -35,7 +35,7 @@ def mock_embed_func(input_data):
def test_with_embeddings():
for wrap_api in [True, False]:
- if wrap_api and sys.version_info.minor >= 11:
+ if wrap_api:
# ratelimiter package doesn't work on 3.11
continue
data = pa.Table.from_arrays(