Skip to content

Commit

Permalink
formatting fix
Browse files Browse the repository at this point in the history
  • Loading branch information
firojalam committed Feb 3, 2024
1 parent ee9500b commit 9c27699
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ def config():
"model_args": {
"max_tries": 3,
},
"general_args": {"test_split": "task1A", "train_split":"task1A"},
"general_args": {"test_split": "task1A", "train_split": "task1A"},
}


Expand Down
1 change: 0 additions & 1 deletion llmebench/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,6 @@ def get_data_paths(config, split):
dataset_args = config.get("dataset_args", {})
dataset = config["dataset"](**dataset_args)


if split == "test":
data_args = config.get("general_args", {})
elif split == "train":
Expand Down

0 comments on commit 9c27699

Please sign in to comment.