use pre-commit

This commit is contained in:
hiyouga committed 2024-10-29 09:07:46 +00:00
1 parent 77666bd227
commit 21db8ed2f4
86 files changed
+1048 -1064

No files matched your search

+1 -1
View File
@@ -61,7 +61,7 @@ OS_NAME = os.environ.get("OS_NAME", "")
],
)
def test_run_exp(stage: str, dataset: str):
output_dir = "train_{}".format(stage)
output_dir = f"train_{stage}"
run_exp({"stage": stage, "dataset": dataset, "output_dir": output_dir, **TRAIN_ARGS})
assert os.path.exists(output_dir)