From f5669c10a54c2f214ceae5044247b0c07739814e Mon Sep 17 00:00:00 2001
From: Setepenre <pierre.delaunay.tr@gmail.com>
Date: Wed, 3 Jul 2024 12:34:52 -0400
Subject: [PATCH] Ensure that torch metadata gathering cannot fail

---
 milabench/metadata.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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())