match ray cluster python version

bioimage-io · Dec 14, 2024 · 000097d · 000097d
1 parent 246b2a9
commit 000097d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -1,5 +1,5 @@
 # Use an image with Python 3.11
-FROM python:3.11-slim
+FROM python:3.11.9-slim
 
 # Set the working directory
 WORKDIR /app/

diff --git a/bioimageio_colab/register_sam_service.py b/bioimageio_colab/register_sam_service.py
@@ -224,6 +224,8 @@ def compute_mask_function(**kwargs: dict):
         def test_model_function(**kwargs: dict):
             kwargs["cache_dir"] = cache_dir
             return ray.get(test_model_ray.remote(kwargs))
+
+        logger.info(f"Successfully connected to Ray (version: {ray.__version__})")
 
     else:
         compute_embedding_function = partial(