Skip to content

GitLab

Unverified Commit faa6bebe authored by Yuge Zhang's avatar Yuge Zhang Committed by GitHub
Browse files

Add missing reference of state dict hook (#4764)

parent e0ce406f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
nni/retiarii/utils.py
View file @ faa6bebe
...@@ -11,7 +11,7 @@ from pathlib import Path ...@@ -11,7 +11,7 @@ from pathlib import Path
from nni.common.hpo_utils import ParameterSpec from nni.common.hpo_utils import ParameterSpec
__all__ = ['NoContextError', 'ContextStack', 'ModelNamespace'] __all__ = ['NoContextError', 'ContextStack', 'ModelNamespace', 'original_state_dict_hooks']
def import_(target: str, allow_none: bool = False) -> Any: def import_(target: str, allow_none: bool = False) -> Any:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment