fix: coop property

SwanHubX · Nov 18, 2024 · 74642eb · 74642eb
1 parent 82202bd
commit 74642eb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/swanlab/data/run/metadata/cooperation.py b/swanlab/data/run/metadata/cooperation.py
@@ -15,10 +15,10 @@
 
 
 def get_cooperation_info():
-    qin_cloud = get_qing_cloud_info()
+    qing_cloud = get_qing_cloud_info()
     coop = {"swanlab": get_swanlab_info()}
-    if qin_cloud:
-        coop.update(qin_cloud)
+    if qing_cloud:
+        coop.update({"qing_cloud": qing_cloud})
     return coop
 
 

diff --git a/swanlab/data/run/metadata/hardware.py b/swanlab/data/run/metadata/hardware.py
@@ -155,7 +155,7 @@ def get_cuda_version():
             # 获取 GPU 的总显存, 单位为GB
             info["memory"].append(round(pynvml.nvmlDeviceGetMemoryInfo(handle).total / (1024**3)))
 
-    except pynvml.NVMLError as e:
+    except pynvml.NVMLError:
         pass
     finally:
         # 结束 NVML