[diffusion] CI: fix breaking import path in nightly (#21449)

scripts/ci/utils/diffusion/publish_comparison_results.py

@@ -20,19 +20,33 @@ import time
 from datetime import datetime, timezone
 from pathlib import Path
 
-# Reuse GitHub API helpers from publish_traces
-sys.path.insert(0, str(Path(__file__).parent))
-from ..publish_traces import (
-    create_blobs,
-    create_commit,
-    create_tree,
-    get_branch_sha,
-    get_tree_sha,
-    is_permission_error,
-    is_rate_limit_error,
-    update_branch_ref,
-    verify_token_permissions,
-)
+# Reuse GitHub API helpers from publish_traces.
+# Support both direct script execution and package-style imports.
+if __package__:
+    from ..publish_traces import (
+        create_blobs,
+        create_commit,
+        create_tree,
+        get_branch_sha,
+        get_tree_sha,
+        is_permission_error,
+        is_rate_limit_error,
+        update_branch_ref,
+        verify_token_permissions,
+    )
+else:
+    sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+    from publish_traces import (
+        create_blobs,
+        create_commit,
+        create_tree,
+        get_branch_sha,
+        get_tree_sha,
+        is_permission_error,
+        is_rate_limit_error,
+        update_branch_ref,
+        verify_token_permissions,
+    )
 
 # Repository configuration
 REPO_OWNER = "sglang-bot"