Skip to content

Commit 0b24150

Browse files
scripts: update qdc pytest script to make linter happy
1 parent 5a4955b commit 0b24150

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

scripts/snapdragon/qdc/tests/test_bench.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
1-
import pytest,subprocess,sys
1+
import pytest
2+
import subprocess
3+
import sys
24

35
tmp_path='/data/local/tmp'
46
pkg_path=f'{tmp_path}/llama.cpp'
@@ -23,10 +25,9 @@ def test_install():
2325

2426
## Basic cli tests
2527
def run_llama_cli(dev, opts):
26-
run_cmd([ 'adb', 'shell',
27-
f'{cli_pref}/llama-cli -m {model} --device {dev} -ngl 99 --batch-size 128 -n 128 -t 4 -no-cnv --seed 42 {opts} ' +
28-
f'-p "what is the most popular cookie in the world?\nPlease provide a very brief bullet point summary.\nBegin your answer with **BEGIN**."'
29-
])
28+
prompt='what is the most popular cookie in the world?\nPlease provide a very brief bullet point summary.\nBegin your answer with **BEGIN**.'
29+
opts = '--batch-size 128 -n 128 -no-cnv --seed 42 ' + opts
30+
run_cmd(['adb', 'shell', f'{cli_pref}/llama-cli -m {model} --device {dev} -ngl 99 -t 4 {opts} -p "{prompt}"'])
3031

3132

3233
@pytest.mark.dependency(depends=['test_install'])

0 commit comments

Comments
 (0)