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 (