From 8e854742ae8af0d88d94bdc54312a5ee134b3096 Mon Sep 17 00:00:00 2001 From: Richard Date: Mon, 21 Nov 2022 19:31:58 -0800 Subject: [PATCH] rewrite main cli with proper argument parsing --- neuralprophet/__main__.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/neuralprophet/__main__.py b/neuralprophet/__main__.py index ed92af93e..74006113d 100644 --- a/neuralprophet/__main__.py +++ b/neuralprophet/__main__.py @@ -1,8 +1,13 @@ """ Invokes neuralprophet when module is run as a script. """ - +import argparse from neuralprophet._version import __version__ + +parser = argparse.ArgumentParser(description="NeuralProphet") +parser.add_argument("-v", "--version", action="version", version="%(prog)s " + __version__) + + if __name__ == "__main__": - print(__version__) + parser.parse_args()