kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-13 22:18:36 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix for generate lockfile pipeline

Browse Source 
Signed-off-by: Yuanjing Xue <197832395+yuanjingx87@users.noreply.github.com>
This commit is contained in:
Yuanjing Xue 2025-09-17 14:52:50 -07:00
parent 3b7f83482d
commit 90e53f061e
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
jenkins/GenerateLock.groovy
View File

@ -47,7 +47,7 @@ def generate()
sh "apt update"
sh "apt install -y python3-dev git curl"
sh "git config --global --add safe.directory ${env.WORKSPACE}"
sh "git config --global user.email \"tensorrt_llm@nvidia.com\""
sh "git config --global user.email \"90828364+tensorrt-cicd@users.noreply.github.com\""
sh "git config --global user.name \"TensorRT LLM\""
trtllm_utils.checkoutSource(LLM_REPO, params.llmBranch, env.WORKSPACE, false, false)
sh "python3 --version"
@ -66,6 +66,9 @@ def generate()
withCredentials([usernamePassword(credentialsId: 'github-cred-trtllm-ci', usernameVariable: 'GIT_USER', passwordVariable: 'GIT_PASS')]) {
def authedUrl = LLM_REPO.replaceFirst('https://', "https://${GIT_USER}:${GIT_PASS}@")
sh "git remote set-url origin ${authedUrl}"
sh "git status"
sh "git stash"
sh "git pull --rebase origin ${params.llmBranch}"
sh "git push origin HEAD:${params.llmBranch}"
}
}

2
scripts/generate_lock_file.py
View File

@ -108,7 +108,7 @@ if __name__ == "__main__":
print(f"Initializing PyProject.toml in {file_path}")
project_info = get_project_info(file_path)
name = project_info["name"]
author = '"TensorRT LLM [svc_tensorrt_llm@nvidia.com]"'
author = '"TensorRT LLM [90828364+tensorrt-cicd@users.noreply.github.com]"'
version = project_info["version"]
py_version = '">=3.10,<3.13"'
poetry_init_cmd = f'poetry init --no-interaction --name {name} --author {author} --python {py_version}'