diff --git a/milabench/metadata.py b/milabench/metadata.py index 24733d1bb..d28005542 100644 --- a/milabench/metadata.py +++ b/milabench/metadata.py @@ -11,7 +11,7 @@ from ._version import __commit__, __date__, __tag__ from .scripts.vcs import retrieve_git_versions - +from .utils import error_guard def _get_gpu_info(): try: @@ -21,6 +21,7 @@ def _get_gpu_info(): return {} +@error_guard({}) def fetch_torch_version(pack): cwd = pack.dirs.code exec_env = pack.full_env(dict())