From 0a1fdcd8ebaebf48d95d795c3693a0148f3ec0f9 Mon Sep 17 00:00:00 2001 From: Leander van Eekelen <47320151+leandervaneekelen@users.noreply.github.com> Date: Tue, 6 Dec 2022 23:48:17 +0100 Subject: [PATCH] Add catch for misspelled `--task` (#10420) * Add catch for misspelled task Signed-off-by: Leander van Eekelen <47320151+leandervaneekelen@users.noreply.github.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update val.py Signed-off-by: Glenn Jocher * Update val.py Signed-off-by: Glenn Jocher Signed-off-by: Leander van Eekelen <47320151+leandervaneekelen@users.noreply.github.com> Signed-off-by: Glenn Jocher Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Glenn Jocher --- segment/val.py | 2 ++ val.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/segment/val.py b/segment/val.py index 368a058..5cf8ae8 100644 --- a/segment/val.py +++ b/segment/val.py @@ -463,6 +463,8 @@ def main(opt): np.savetxt(f, y, fmt='%10.4g') # save os.system('zip -r study.zip study_*.txt') plot_val_study(x=x) # plot + else: + raise NotImplementedError(f'--task {opt.task} not in ("train", "val", "test", "speed", "study")') if __name__ == "__main__": diff --git a/val.py b/val.py index e84249e..8d27d9d 100644 --- a/val.py +++ b/val.py @@ -399,6 +399,8 @@ def main(opt): np.savetxt(f, y, fmt='%10.4g') # save os.system('zip -r study.zip study_*.txt') plot_val_study(x=x) # plot + else: + raise NotImplementedError(f'--task {opt.task} not in ("train", "val", "test", "speed", "study")') if __name__ == "__main__":