Reduce printed error when making reports

mila-iqia · Jun 26, 2024 · ac8646a · ac8646a
1 parent 5bb8e13
commit ac8646a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/milabench/commands/__init__.py b/milabench/commands/__init__.py
@@ -659,7 +659,7 @@ def _argv(self, **_) -> List:
             else ["--multi_gpu"]
         )
 
-        cpu_per_process = self.resolve_argument('--cpus_per_gpu')
+        cpu_per_process = self.pack.resolve_argument('--cpus_per_gpu')
         return [
             # -- Run the command in the right venv
             # This could be inside the SSH Command

diff --git a/milabench/report.py b/milabench/report.py
@@ -33,7 +33,7 @@ def _make_row(summary, compare, config, query=None):
         "enabled": is_enabled,
     }
 
-    if not summary:
+    if not summary or summary.get("empty", False):
         return row
 
     # Count not running an enabled benchmark as a failure
@@ -218,7 +218,8 @@ def make_dataframe(summary, compare=None, weights=None, query=None):
                     "n": 0, 
                     "successes": 0,
                     "failures": 0, 
-                    "enabled": weights[key]["enabled"]
+                    "enabled": weights[key]["enabled"],
+                    "empty": True
                 }
 
     if weights is None: