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()