From 03765e1aca7ad3f8aa8de3ca812f3cd1dcdcced2 Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Wed, 13 Jul 2022 12:45:11 +0200 Subject: [PATCH] Fix parallelism not working --- great_ai/__main__.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/great_ai/__main__.py b/great_ai/__main__.py index 777bdb5..3cc12be 100644 --- a/great_ai/__main__.py +++ b/great_ai/__main__.py @@ -10,7 +10,7 @@ from typing import Optional import uvicorn from uvicorn._subprocess import get_subprocess from uvicorn.config import LOGGING_CONFIG, Config -from uvicorn.supervisors.basereload import BaseReload +from uvicorn.supervisors import BaseReload, Multiprocess from watchdog.events import FileSystemEvent, PatternMatchingEventHandler from watchdog.observers import Observer @@ -68,15 +68,13 @@ def serve() -> None: config = Config(app, **common_config) socket = config.bind_socket() - server = GreatAIReload( - config, target=uvicorn.Server(config=config).run, sockets=[socket] - ) - server.startup() try: - Event().wait() + Multiprocess( + config, target=uvicorn.Server(config=config).run, sockets=[socket] + ).run() + finally: - server.shutdown() if args.file_name.endswith(".ipynb"): Path(get_script_name_of_notebook(args.file_name)).unlink( missing_ok=True