add in default value for sysinfo

aidatatools · Apr 15, 2024 · 6afcdfc · 6afcdfc
1 parent 79e7489
commit 6afcdfc
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/llm_benchmark/systeminfo/sysmain.py b/llm_benchmark/systeminfo/sysmain.py
@@ -101,6 +101,10 @@ def get_extra():
     ans={}
     ans['system'] = f"{system_info.system}"
     ans['memory'] = get_total_memory_size()
+    ans['cpu'] = f"unknown"
+    ans['gpu'] = f"unknown"
+    ans['os_version'] = f"unknown"
+
     try:
         if(system_info.system=='Darwin'):
             ans['system_name'] = "macOS"
@@ -213,7 +217,9 @@ def get_extra():
         return ans
 
     except:
-        print("error!")
+        print("error! when retrieving cpu, gpu, os_version")
+
+    return ans
 
 def get_uuid():
     sys_info = get_extra()

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "llm_benchmark"
-version = "0.3.15"
+version = "0.3.16"
 description = "LLM Benchmark for Throughputs via Ollama"
 authors = ["Jason Chuang <[email protected]>"]
 license = "MIT"

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name='llm_benchmark',
-    version='0.3.15',
+    version='0.3.16',
     author='Jason Chuang',
     author_email='[email protected]',
     description='LLM Benchmark',