From 49b9c3be8097c7651cb54a4b2dbeaf3b11b066de Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sun, 19 Mar 2023 21:21:19 -0700 Subject: [PATCH] Include pretrained tag in deprecated mapping warning --- timm/models/_registry.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/timm/models/_registry.py b/timm/models/_registry.py index 532252ed..f089e579 100644 --- a/timm/models/_registry.py +++ b/timm/models/_registry.py @@ -133,7 +133,8 @@ def register_model(fn: Callable[..., Any]) -> Callable[..., Any]: def _deprecated_model_shim(deprecated_name: str, current_fn: Callable = None, current_tag: str = ''): def _fn(pretrained=False, **kwargs): assert current_fn is not None, f'Model {deprecated_name} has been removed with no replacement.' - warnings.warn(f'Mapping deprecated model {deprecated_name} to current {current_fn.__name__}', stacklevel=2) + current_name = '.'.join([current_fn.__name__, current_tag]) if current_tag else current_fn.__name__ + warnings.warn(f'Mapping deprecated model {deprecated_name} to current {current_name}.', stacklevel=2) pretrained_cfg = kwargs.pop('pretrained_cfg', None) return current_fn(pretrained=pretrained, pretrained_cfg=pretrained_cfg or current_tag, **kwargs) return _fn