@@ -326,7 +326,7 @@ def __init__(self, tool: str = "llama-bench"):
326326
327327 # Set table name and schema based on tool
328328 if self .tool == "llama-bench" :
329- self .table_name = "test "
329+ self .table_name = "llama_bench "
330330 db_fields = LLAMA_BENCH_DB_FIELDS
331331 db_types = LLAMA_BENCH_DB_TYPES
332332 elif self .tool == "test-backend-ops" :
@@ -409,17 +409,17 @@ def __init__(self, data_file: str, tool: Any):
409409
410410 # Tool selection logic
411411 if tool is None :
412- if "test " in table_names :
413- self .table_name = "test "
412+ if "llama_bench " in table_names :
413+ self .table_name = "llama_bench "
414414 self .tool = "llama-bench"
415415 elif "test_backend_ops" in table_names :
416416 self .table_name = "test_backend_ops"
417417 self .tool = "test-backend-ops"
418418 else :
419419 raise RuntimeError (f"No suitable table found in database. Available tables: { table_names } " )
420420 elif tool == "llama-bench" :
421- if "test " in table_names :
422- self .table_name = "test "
421+ if "llama_bench " in table_names :
422+ self .table_name = "llama_bench "
423423 self .tool = "llama-bench"
424424 else :
425425 raise RuntimeError (f"Table 'test' not found for tool 'llama-bench'. Available tables: { table_names } " )
0 commit comments