fixes

scalableminds · Sep 26, 2023 · 31f32ca · 31f32ca
1 parent c5a61eb
commit 31f32ca
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 25 deletions.
diff --git a/cluster_tools/cluster_tools/executors/dask.py b/cluster_tools/cluster_tools/executors/dask.py
@@ -20,7 +20,7 @@
 from cluster_tools.executors.multiprocessing_ import CFutDict, MultiprocessingExecutor
 
 if TYPE_CHECKING:
-    from dask.distributed import Client
+    from distributed import Client
 
 _T = TypeVar("_T")
 _P = ParamSpec("_P")
@@ -34,13 +34,13 @@ def __init__(
         self,
         **kwargs: Any,
     ) -> None:
-        from dask.distributed import Client
+        from distributed import Client
 
         self.client = Client(**kwargs)
 
     @classmethod
     def as_completed(cls, futures: List["Future[_T]"]) -> Iterator["Future[_T]"]:
-        from dask.distributed import as_completed
+        from distributed import as_completed
 
         return as_completed(futures)
 

diff --git a/cluster_tools/poetry.lock b/cluster_tools/poetry.lock
diff --git a/cluster_tools/pyproject.toml b/cluster_tools/pyproject.toml
@@ -9,7 +9,7 @@ repository = "https://github.com/scalableminds/webknossos-libs"
 
 [tool.poetry.dependencies]
 python = ">=3.8"
-dask = { version = "^2023.9.1", python = ">=3.9", extras = ["distributed"] }
+distributed = { version = "^2023.9.1", python = ">=3.9" }
 kubernetes = "^27.2.0"
 typing-extensions = "^4.4.0"
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,2 +1,2 @@
-dunamai==1.17.0
-poetry==1.5.1
+dunamai==1.18.1
+poetry==1.6.1