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

scripts/ci/utils/diffusion/publish_comparison_results.py

@@ -20,8 +20,9 @@ import time
 from datetime import datetime, timezone
 from pathlib import Path
 
-# Reuse GitHub API helpers from publish_traces
+# Reuse GitHub API helpers from publish_traces.
-sys.path.insert(0, str(Path(__file__).parent))
+# Support both direct script execution and package-style imports.
+if __package__:
     from ..publish_traces import (
         create_blobs,
         create_commit,
@@ -33,6 +34,19 @@ from ..publish_traces import (
         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"

scripts/ci/utils/diffusion/publish_diffusion_gt.py

@@ -6,10 +6,11 @@ via the GitHub API (same pattern as publish_traces.py).
 import argparse
 import os
 import sys
+from pathlib import Path
 
-# Allow importing from the same directory (scripts/ci/utils/)
+# Reuse GitHub API helpers from publish_traces.
-sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+# Support both direct script execution and package-style imports.
-
+if __package__:
     from ..publish_traces import (
         create_blobs,
         create_commit,
@@ -21,6 +22,19 @@ from ..publish_traces import (
         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,
+    )
 
 REPO_OWNER = "sglang-bot"
 REPO_NAME = "sglang-ci-data"