From 79e0d59eeb0431f4654197584f8864e3c40af7f4 Mon Sep 17 00:00:00 2001 From: Ayush Chaurasia Date: Wed, 14 Feb 2024 00:18:19 +0530 Subject: [PATCH] Pin `duckdb<=0.9.2` to avoid 0.10.0 errors (#8181) Co-authored-by: Glenn Jocher --- pyproject.toml | 3 +-- ultralytics/data/explorer/explorer.py | 3 ++- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index aefb6099..f41a55c4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -104,12 +104,11 @@ export = [ ] explorer = [ "lancedb", # vector search - "duckdb", # SQL queries, supports lancedb tables + "duckdb<=0.9.2", # SQL queries, duckdb==0.10.0 bug https://github.com/ultralytics/ultralytics/pull/8181 "streamlit", # visualizing with GUI ] # tensorflow>=2.4.1,<=2.13.1 # TF exports (-cpu, -aarch64, -macos) # tflite-support # for TFLite model metadata -# scikit-learn==0.19.2 # CoreML quantization # nvidia-pyindex # TensorRT export # nvidia-tensorrt # TensorRT export logging = [ diff --git a/ultralytics/data/explorer/explorer.py b/ultralytics/data/explorer/explorer.py index 6dba618d..d21a5c2e 100644 --- a/ultralytics/data/explorer/explorer.py +++ b/ultralytics/data/explorer/explorer.py @@ -59,7 +59,8 @@ class Explorer: model: str = "yolov8n.pt", uri: str = USER_CONFIG_DIR / "explorer", ) -> None: - checks.check_requirements(["lancedb>=0.4.3", "duckdb"]) + # Note duckdb==0.10.0 bug https://github.com/ultralytics/ultralytics/pull/8181 + checks.check_requirements(["lancedb>=0.4.3", "duckdb<=0.9.2"]) import lancedb self.connection = lancedb.connect(uri)