diff --git a/locust/__main__.py b/locust/__main__.py new file mode 100644 index 0000000000..5d6a8109e6 --- /dev/null +++ b/locust/__main__.py @@ -0,0 +1,3 @@ +from .main import main + +main() diff --git a/locust/main.py b/locust/main.py index f4a81d238c..7362c3c1ec 100644 --- a/locust/main.py +++ b/locust/main.py @@ -510,6 +510,3 @@ def sig_term_handler(): shutdown(code=code) except KeyboardInterrupt as e: shutdown(0) - -if __name__ == '__main__': - main()