From d3d9f658e522cf10aede23dc41bf995313d2b98e Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Mon, 4 Mar 2024 16:26:42 +0100
Subject: [PATCH] parameterize FastAPI app name, to be overridden by external
caller if needed
---
model_server/scripts/run_server.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/model_server/scripts/run_server.py b/model_server/scripts/run_server.py
index ca8126e4..b40ff2b3 100644
--- a/model_server/scripts/run_server.py
+++ b/model_server/scripts/run_server.py
@@ -30,12 +30,12 @@ def parse_args():
return parser.parse_args()
-def main(args) -> None:
+def main(args, app_name='model_server.base.api:app') -> None:
print('CLI args:\n' + str(args))
server_process = Process(
target=uvicorn.run,
- args=('model_server.base.api:app',),
+ args=(app_name,),
kwargs={
'app_dir': '.',
'host': args.host,
--
GitLab