From 5b7fafe2c4a70a4302a9e264aecec45afed218ee Mon Sep 17 00:00:00 2001 From: Nils Mechtel Date: Thu, 19 Dec 2024 02:20:10 +0100 Subject: [PATCH] fix await --- bioimageio_colab/register_sam_service.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bioimageio_colab/register_sam_service.py b/bioimageio_colab/register_sam_service.py index b101831..1b8c460 100644 --- 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