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

This commit is contained in:
Mick
2026-03-26 16:33:22 +08:00
committed by GitHub
parent 35720d9969
commit 238a4b8f8f
2 changed files with 55 additions and 27 deletions
@@ -20,19 +20,33 @@ import time
from datetime import datetime, timezone from datetime import datetime, timezone
from pathlib import Path 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.
from ..publish_traces import ( if __package__:
create_blobs, from ..publish_traces import (
create_commit, create_blobs,
create_tree, create_commit,
get_branch_sha, create_tree,
get_tree_sha, get_branch_sha,
is_permission_error, get_tree_sha,
is_rate_limit_error, is_permission_error,
update_branch_ref, is_rate_limit_error,
verify_token_permissions, 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 # Repository configuration
REPO_OWNER = "sglang-bot" REPO_OWNER = "sglang-bot"
@@ -6,21 +6,35 @@ via the GitHub API (same pattern as publish_traces.py).
import argparse import argparse
import os import os
import sys 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 ( from ..publish_traces import (
create_blobs, create_blobs,
create_commit, create_commit,
create_tree, create_tree,
get_branch_sha, get_branch_sha,
get_tree_sha, get_tree_sha,
is_permission_error, is_permission_error,
is_rate_limit_error, is_rate_limit_error,
update_branch_ref, update_branch_ref,
verify_token_permissions, 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_OWNER = "sglang-bot"
REPO_NAME = "sglang-ci-data" REPO_NAME = "sglang-ci-data"