Resolve "Flavor for compute backend config for local runs"

• DoD is fulfilled
• Acceptance criteria are checked
• CHANGELOG is updated

Closes #203 (closed)

mantik/cli/runs/local.py

@@ -5,8 +5,6 @@ import typing as t
 import uuid
 
 import click
-import GPUtil
-import psutil
 
 import mantik
 import mantik.cli._options as _main_options
@@ -20,17 +18,19 @@ import mantik.utils.env_vars as env_vars
 logger = logging.getLogger(__name__)
 
 
-def get_gpu_info() -> dict:
+def get_gpu_info() -> list:
+    import GPUtil
+
     gpus = GPUtil.getGPUs()
 
     gpu_info_list = []
 
     for gpu in gpus:
         gpu_info = {
-            "name": gpu.name,
-            "id": str(gpu.id),
-            "driver": gpu.driver,
-            "memory_total": str(gpu.memoryTotal),
+            "Name": gpu.name,
+            "ID": str(gpu.id),
+            "Driver": gpu.driver,
+            "TotalMemory": str(gpu.memoryTotal),
         }
 
         gpu_info_list.append(gpu_info)
@@ -40,10 +40,11 @@ def get_gpu_info() -> dict:
 
 def get_system_details() -> dict:
     import cpuinfo
+    import psutil
 
     system_details = {
         "Platform": platform.platform(),
-        "CpuModel": cpuinfo.get_cpu_info()["brand_raw"],  # FIX
+        "CpuModel": cpuinfo.get_cpu_info().get("brand_raw"),
         "CpuCoreCount": psutil.cpu_count(),
         "TotalMemoryGB": psutil.virtual_memory().total / (1024**3),
         "GPUInfo": get_gpu_info(),
@@ -183,9 +184,7 @@ def run_project(
 
     mantik_token = mantik.authentication.auth.get_valid_access_token()
 
-    system_details = {}
-    if backend_config_system_info:
-        system_details = get_system_details()
+    system_details = get_system_details() if backend_config_system_info else {}
 
     local_runs.run(
         data=schemas.RunConfiguration(