diff --git a/scripts/check_all_metrics.py b/scripts/check_all_metrics.py index 45c0c599..fae5720d 100644 --- a/scripts/check_all_metrics.py +++ b/scripts/check_all_metrics.py @@ -59,10 +59,10 @@ def get_model_name(queue, task_id: str): model_name = route.split(".")[4] # model_name = "train_component" return model_name[6:] - # Raise an exception if no matching route was found, this can happens when - # the current task has a dependency to a non-training task or if the route + # Show a warning if no matching route was found, this can happen when the + # current task has a dependency to a non-training task or if the route # pattern changes. - raise Exception(f"No matching route found for task id {task_id}") + LOGGER.warning(f"No matching route found for task id {task_id}") def get_model_names(task_id: str) -> List[str]: @@ -80,8 +80,10 @@ def get_model_names(task_id: str) -> List[str]: ) model_name = get_model_name(queue, task_id) - LOGGER.info("Adding model %r to download list", model_name) - model_names.append(model_name) + + if model_name: + LOGGER.info("Adding model %r to download list", model_name) + model_names.append(model_name) return model_names