From 9a2b2336dcb604833260ee2291a5ed3580521282 Mon Sep 17 00:00:00 2001 From: James Date: Thu, 19 Sep 2024 16:37:38 +0700 Subject: [PATCH] fix: e2e failed test when download model with selection --- engine/e2e-test/test_cli_model_delete.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/engine/e2e-test/test_cli_model_delete.py b/engine/e2e-test/test_cli_model_delete.py index e5373fac1..eda21d1ce 100644 --- a/engine/e2e-test/test_cli_model_delete.py +++ b/engine/e2e-test/test_cli_model_delete.py @@ -1,5 +1,5 @@ import pytest -from test_runner import run +from test_runner import popen, run class TestCliModelDelete: @@ -8,7 +8,7 @@ class TestCliModelDelete: def setup_and_teardown(self): # Setup # Pull model - run("Pull Model", ["pull", "tinyllama"], 120) + stdout, stderr, return_code = popen(["pull", "tinyllama"], "1\n") yield