Ensure that torch metadata gathering cannot fail

mila-iqia · Jul 3, 2024 · f5669c1 · f5669c1
1 parent d669feb
commit f5669c1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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())