From 04025c0efa6b6d9c1c8ca7cf5ae7969347ed72e2 Mon Sep 17 00:00:00 2001 From: Ian Carrasco Date: Tue, 26 Sep 2023 11:10:56 -0700 Subject: [PATCH] Add assertion that at least one base model was specified. Otherwise throw an error message. --- model_tools/check_submission/check_models.py | 1 + 1 file changed, 1 insertion(+) diff --git a/model_tools/check_submission/check_models.py b/model_tools/check_submission/check_models.py index fb05020..20ab0b4 100644 --- a/model_tools/check_submission/check_models.py +++ b/model_tools/check_submission/check_models.py @@ -34,6 +34,7 @@ def check_brain_model_processing(model): def check_base_models(module): module = __import__(module) + assert len(module.get_model_list()) >= 1, 'No base models were specified!' for model in module.get_model_list(): layers = module.get_layers(model) assert layers is not None