diff --git a/conda/environments/all_cuda-121_arch-x86_64.yaml b/conda/environments/all_cuda-121_arch-x86_64.yaml index fdc3194a9c..3ff66690dc 100644 --- a/conda/environments/all_cuda-121_arch-x86_64.yaml +++ b/conda/environments/all_cuda-121_arch-x86_64.yaml @@ -117,6 +117,7 @@ dependencies: - --find-links https://data.dgl.ai/wheels-test/repo.html - --find-links https://data.dgl.ai/wheels/cu121/repo.html - PyMuPDF==1.23.21 + - databricks-cli < 0.100 - databricks-connect - dgl==2.0.0 - dglgo diff --git a/conda/environments/dev_cuda-121_arch-x86_64.yaml b/conda/environments/dev_cuda-121_arch-x86_64.yaml index da84bab2ae..caea586cd3 100644 --- a/conda/environments/dev_cuda-121_arch-x86_64.yaml +++ b/conda/environments/dev_cuda-121_arch-x86_64.yaml @@ -95,6 +95,7 @@ dependencies: - zlib=1.2.13 - pip: - PyMuPDF==1.23.21 + - databricks-cli < 0.100 - databricks-connect - milvus==2.3.5 - pymilvus==2.3.6 diff --git a/conda/environments/examples_cuda-121_arch-x86_64.yaml b/conda/environments/examples_cuda-121_arch-x86_64.yaml index 2b663d6b1a..e2e32c67a1 100644 --- a/conda/environments/examples_cuda-121_arch-x86_64.yaml +++ b/conda/environments/examples_cuda-121_arch-x86_64.yaml @@ -61,6 +61,7 @@ dependencies: - --find-links https://data.dgl.ai/wheels-test/repo.html - --find-links https://data.dgl.ai/wheels/cu121/repo.html - PyMuPDF==1.23.21 + - databricks-cli < 0.100 - databricks-connect - dgl==2.0.0 - dglgo diff --git a/conda/environments/runtime_cuda-121_arch-x86_64.yaml b/conda/environments/runtime_cuda-121_arch-x86_64.yaml index 791f58e463..2c5a21bdf3 100644 --- a/conda/environments/runtime_cuda-121_arch-x86_64.yaml +++ b/conda/environments/runtime_cuda-121_arch-x86_64.yaml @@ -37,6 +37,7 @@ dependencies: - watchdog=3.0 - websockets - pip: + - databricks-cli < 0.100 - databricks-connect - milvus==2.3.5 - pymilvus==2.3.6 diff --git a/dependencies.yaml b/dependencies.yaml index 7f074356e2..eb2e655cba 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -271,6 +271,7 @@ dependencies: - websockets - pip - pip: + - databricks-cli < 0.100 - databricks-connect - milvus==2.3.5 # update to match pymilvus when available - pymilvus==2.3.6