From 000097d6510cd9afd9d98775a7f96e4ec645f0ee Mon Sep 17 00:00:00 2001
From: Nils Mechtel <nils.mech@gmail.com>
Date: Sun, 15 Dec 2024 00:09:22 +0100
Subject: [PATCH] match ray cluster python version

---
 Dockerfile                               | 2 +-
 bioimageio_colab/register_sam_service.py | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index ca0dc7a..486e0c7 100644
--- 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
index 70e951c..e0df310 100644
--- 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(