Skip to content

Commit 5789b79

Browse files
committed
Fix test to adjust GPU as well
1 parent 6252d1c commit 5789b79

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/flow/tests_commands.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -557,7 +557,8 @@ def test_ai_config(env):
557557
env.assertEqual(res, 511*1024*1024)
558558

559559
# Change the default backends path and load backend.
560-
con.execute_command('AI.CONFIG', 'BACKENDSPATH', ROOT+"/install-cpu")
560+
path = f'{ROOT}/install-{DEVICE.lower()}'
561+
con.execute_command('AI.CONFIG', 'BACKENDSPATH', path)
561562
res = con.execute_command('AI.CONFIG', 'GET', 'BACKENDSPATH')
562563
env.assertEqual(res, f'{ROOT}/install-cpu'.encode())
563564
be_info = get_info_section(con, "backends_info")

0 commit comments

Comments
 (0)