Skip to content

Commit 10e1b85

Browse files
authored
fix(e2e): correct ml-model-selection download output-dir path after training dir restructure (vllm-project#1389)
Signed-off-by: bitliu <bitliu@tencent.com>
1 parent dc80ef9 commit 10e1b85

File tree

2 files changed

+2
-199
lines changed

2 files changed

+2
-199
lines changed

e2e/profiles/ml-model-selection/profile.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -366,7 +366,7 @@ func (p *Profile) prepareMLModels(ctx context.Context) error {
366366
os.MkdirAll(sourceDir, 0755)
367367

368368
downloadCmd := exec.CommandContext(ctx, "python3", "download_model.py",
369-
"--output-dir", "../../../.cache/ml-models",
369+
"--output-dir", "../../../../.cache/ml-models",
370370
"--repo-id", "abdallah1008/semantic-router-ml-models",
371371
)
372372
downloadCmd.Dir = trainingDir
@@ -376,7 +376,7 @@ func (p *Profile) prepareMLModels(ctx context.Context) error {
376376
if err := downloadCmd.Run(); err != nil {
377377
// Try with python if python3 fails
378378
downloadCmd = exec.CommandContext(ctx, "python", "download_model.py",
379-
"--output-dir", "../../../.cache/ml-models",
379+
"--output-dir", "../../../../.cache/ml-models",
380380
"--repo-id", "abdallah1008/semantic-router-ml-models",
381381
)
382382
downloadCmd.Dir = trainingDir

src/semantic-router/pkg/imagegen/integration_test.go

Lines changed: 0 additions & 197 deletions
This file was deleted.

0 commit comments

Comments
 (0)