[Misc] clean Temporary CI Configs (#2784)

Signed-off-by: neptune <neptune@hust.edu.cn>
Co-authored-by: neptune <neptune@hust.edu.cn>
This commit is contained in:
n1ptuneandneptune authored and GitHub committed 2026-04-14 12:41:23 +00:00
1 parent f87674aa44
commit bcd5f16321
1 file changed
+14 -11
+14 -11
View File
@@ -1,3 +1,4 @@
import atexit
import base64
import datetime
import io
@@ -1362,9 +1363,10 @@ def modify_stage_config(
continue
# Delete specified paths in this stage
for path in delete_paths:
if path: # Skip empty paths
delete_by_path(target_stage, path)
# Avoid shadowing the original YAML Path used for the output filename below.
for delete_path in delete_paths:
if delete_path: # Skip empty paths
delete_by_path(target_stage, delete_path)
elif "." in key:
# Delete using dot-separated path
delete_by_path(config, key)
@@ -1394,15 +1396,15 @@ def modify_stage_config(
raise KeyError(f"Stage ID {stage_id} not found, available: {available_ids}")
# Apply updates to this stage
for path, val in stage_updates.items():
for update_path, val in stage_updates.items():
# Check if this is a simple key (not dot-separated)
# Example: 'engine_input_source' vs 'engine_args.max_model_len'
if "." not in path:
if "." not in update_path:
# Direct key assignment (e.g., updating a list value)
target_stage[path] = val
target_stage[update_path] = val
else:
# Dot-separated path (e.g., nested dict access)
apply_update(target_stage, path, val)
apply_update(target_stage, update_path, val)
elif "." in key:
# Apply using dot-separated path
apply_update(config, key, value)
@@ -1414,13 +1416,14 @@ def modify_stage_config(
# within the same second (e.g. test_qwen3_omni_expansion imports both
# get_chunk_config and get_batch_token_config). int(time.time()) would collide
# and the later write would overwrite the earlier YAML on disk.
base_name = yaml_path.rsplit(".", 1)[0] if "." in yaml_path else yaml_path
output_path = f"{base_name}_{time.time_ns()}.yaml"
# Keep generated configs outside the repo and delete them when pytest exits.
output_fd, output_path = tempfile.mkstemp(prefix=f"{path.stem}_", suffix=".yaml")
atexit.register(Path(output_path).unlink, missing_ok=True)
with open(output_path, "w", encoding="utf-8") as f:
with os.fdopen(output_fd, "w", encoding="utf-8") as f:
yaml.dump(config, f, default_flow_style=None, sort_keys=False, allow_unicode=True, indent=2)
return output_path
return str(output_path)
class OmniServer: