fix:callback (#656)

* fix:callback * fix:crashed info * update version
SwanHubX · Jul 26, 2024 · 304bc54 · 304bc54
1 parent 29f4d35
commit 304bc54
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/swanlab/cli/commands/task/search.py b/swanlab/cli/commands/task/search.py
@@ -52,4 +52,4 @@ def search(cuid):
     console.print(f"[bold]Created At:[/bold] {tm.created_at}")
     tm.started_at is not None and console.print(f"[bold]Started At:[/bold] {tm.started_at}")
     tm.finished_at is not None and console.print(f"[bold]Finished At:[/bold] {tm.finished_at}")
-    tm.msg is not None and console.print(f"[bold][red]Task Error[/red]:[/bold] \n\n{tm.msg}\n")
+    tm.status == 'CRASHED' and console.print(f"[bold][red]Task Error[/red]:[/bold] \n\n{tm.msg}\n")
diff --git a/swanlab/data/callback_cloud.py b/swanlab/data/callback_cloud.py
@@ -230,7 +230,7 @@ def _write_call_call(message):
             show_button_html(experiment_url)
 
         # task环境下，同步实验信息回调
-        if SwanLabEnv.RUNTIME.value == "task":
+        if os.environ.get(SwanLabEnv.RUNTIME.value) == "task":
             cuid = os.environ["SWANLAB_TASK_ID"]
             info = {
                 "cuid": cuid,

diff --git a/swanlab/package.json b/swanlab/package.json
@@ -1,6 +1,6 @@
 {
   "name": "swanlab",
-  "version": "0.3.15-alpha.1",
+  "version": "0.3.15-alpha.2",
   "description": "",
   "python": "true"
 }