From 8f72fd333d6eeffb34b38e441dfcea7f6eb1698d Mon Sep 17 00:00:00 2001 From: Vanessa Mac Date: Thu, 19 Dec 2024 17:35:05 -0500 Subject: [PATCH] Fail the job if no backends are online --- ami/ml/models/pipeline.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/ami/ml/models/pipeline.py b/ami/ml/models/pipeline.py index ce34b6f7..b2feae17 100644 --- a/ami/ml/models/pipeline.py +++ b/ami/ml/models/pipeline.py @@ -486,9 +486,10 @@ def choose_backend_for_pipeline(self, job_id): return backend_id_lowest_latency def process_images(self, images: typing.Iterable[SourceImage], job_id: int | None = None): - backend_id = self.choose_backend_for_pipeline(job_id) - if not isinstance(backend_id, int): - return backend_id + try: + backend_id = self.choose_backend_for_pipeline(job_id) + except Exception: + return if not self.backends.filter(pk=backend_id).first().endpoint_url: raise ValueError("No endpoint URL configured for this pipeline")