fix await

bioimage-io · Dec 19, 2024 · 5b7fafe · 5b7fafe
1 parent 6ec00e8
commit 5b7fafe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bioimageio_colab/register_sam_service.py b/bioimageio_colab/register_sam_service.py
@@ -141,7 +141,7 @@ async def test_model(handles: dict, model_name: str, context: dict = None) -> di
     logger.info(f"User '{user_id}' - Test run for model '{model_name}'...")
 
     image = np.random.rand(1024, 1024)
-    result  = compute_image_embedding(handles, image, model_name, context)
+    result = await compute_image_embedding(handles, image, model_name, context)
 
     assert "features" in result
     assert "input_size" in result