From 42c55a338c95d813c77b6fef9a87a58fa32b4d1d Mon Sep 17 00:00:00 2001 From: Stephen Shao Date: Mon, 2 Jun 2025 21:58:54 -0400 Subject: [PATCH] Added logic to check model is deprecated or not --- src/madengine/tools/run_models.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/madengine/tools/run_models.py b/src/madengine/tools/run_models.py index dd528c4e..f9832e4e 100644 --- a/src/madengine/tools/run_models.py +++ b/src/madengine/tools/run_models.py @@ -905,6 +905,13 @@ def run_model(self, model_info: typing.Dict) -> bool: # Environment variable updates for MAD Public CI run_details.gpu_architecture = self.context.ctx["docker_env_vars"]["MAD_SYSTEM_GPU_ARCHITECTURE"] + # Check if model is deprecated + if model_info.get("is_deprecated", False): + print(f"WARNING: Model {model_info['name']} has been deprecated.") + if self.args.skip_deprecated_models: + print(f"Skipping deprecated model {model_info['name']}") + return True # Return success to not affect overall status + # check if model is supported on current gpu architecture, if not skip. list_skip_gpu_arch = [] if (