Skip to content
Snippets Groups Projects
Commit 5b9c5dd4 authored by simon wu's avatar simon wu Committed by Kai Chen
Browse files

Require an operation(Save or Show the results) when testing (#993)

* Require an operation(Save or Show the results) when testing

* using parentheses
parent 66449a50
No related branches found
No related tags found
No related merge requests found
......@@ -128,6 +128,10 @@ def parse_args():
def main():
args = parse_args()
assert args.out or args.show, \
('Please specify at least one operation (save or show the results) '
'with the argument "--out" or "--show"')
if args.out is not None and not args.out.endswith(('.pkl', '.pickle')):
raise ValueError('The output file must be a pkl file.')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment