Skip to content

Commit d94954c

Browse files
committed
More fix to the logic of data source type check
1 parent a6aff9d commit d94954c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

genai_bench/data/config.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,8 @@ def from_cli_args(
102102
file_format = "txt"
103103
elif path.suffix == ".json":
104104
file_format = "json"
105+
else:
106+
raise ValueError(f"Unsupported file format: {path.suffix}")
105107
else:
106108
# Assume it's a HuggingFace ID
107109
source_type = "huggingface"

0 commit comments

Comments
 (0)