diff --git a/tests/test_api.py b/tests/test_api.py index 8489688afccc45a59386e68670204e9913f16245..2b67a67d84e4df26c7f6432070845104f81ca9f1 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -7,6 +7,8 @@ from model_server.conf.testing import czifile class TestServerBaseClass(unittest.TestCase): + app_name = 'model_server.base.api:app' + def setUp(self) -> None: import uvicorn host = '127.0.0.1' @@ -14,7 +16,7 @@ class TestServerBaseClass(unittest.TestCase): self.server_process = Process( target=uvicorn.run, - args=('model_server.base.api:app', ), + args=(self.app_name, ), kwargs={'host': host, 'port': port, 'log_level': 'debug'}, daemon=True )