Skip to content

GitLab

Commit 4ff8260d authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

allow for trailing comma in model_args

parent da43b3a2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/utils.py
View file @ 4ff8260d
...@@ -54,7 +54,7 @@ def simple_parse_args_string(args_string): ...@@ -54,7 +54,7 @@ def simple_parse_args_string(args_string):
args_string = args_string.strip() args_string = args_string.strip()
if not args_string: if not args_string:
return {} return {}
arg_list = args_string.split(",") arg_list = [arg for arg in args_string.split(",") if arg]
args_dict = {k: v for k, v in [arg.split("=") for arg in arg_list]} args_dict = {k: v for k, v in [arg.split("=") for arg in arg_list]}
return args_dict return args_dict
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment