Add key for system metadata for benchmarks (#404)

pytorch · Nov 20, 2024 · 789cead · 789cead
1 parent afc1647
commit 789cead
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/benchmarks/decoders/generate_readme_data.py b/benchmarks/decoders/generate_readme_data.py
@@ -85,12 +85,13 @@ def main() -> None:
     )
     df_data.append(
         {
-            "cpu_count": os.cpu_count(),
-            "system": platform.system(),
-            "machine": platform.machine(),
-            "processor": platform.processor(),
-            "python_version": str(platform.python_version()),
-            "is_cuda_available": str(torch.cuda.is_available()),
+            "system_metadata": {
+                "cpu_count": os.cpu_count(),
+                "system": platform.system(),
+                "machine": platform.machine(),
+                "python_version": str(platform.python_version()),
+                "is_cuda_available": str(torch.cuda.is_available()),
+            }
         }
     )