fix #112 - don't subset cpuinfo

DeepLabCut · Feb 14, 2024 · cf4cef3 · cf4cef3
1 parent 427c126
commit cf4cef3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dlclive/benchmark.py b/dlclive/benchmark.py
@@ -130,7 +130,7 @@ def get_system_info() -> dict:
     else:
         from cpuinfo import get_cpu_info
 
-        dev = [get_cpu_info()["brand"]]
+        dev = get_cpu_info()
         dev_type = "CPU"
 
     return {