Update src/fondant/components/segment_images/src/main.py

Co-authored-by: Robbe Sneyders <[email protected]>
ml6team · Mar 13, 2024 · c3a1e78 · c3a1e78
1 parent fb55154
commit c3a1e78
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/fondant/components/segment_images/src/main.py b/src/fondant/components/segment_images/src/main.py
@@ -131,6 +131,9 @@ def __init__(
 
     def setup(self) -> Client:
         """Setup LocalCudaCluster if gpu is available."""
+        dask.config.set({"dataframe.convert-string": False})
+        dask.config.set({"distributed.worker.daemon": False})
+
         if self.device == "cuda":
             cluster = LocalCUDACluster()
             return Client(cluster)