fix tests

Author: technillogue

python/tests/server/test_runner.py

@@ -43,7 +43,7 @@ async def runner():
         await runner.setup()
         yield runner
     finally:
-        runner.shutdown()
+        await runner.shutdown()
 
 
 @pytest_asyncio.fixture
@@ -60,7 +60,7 @@ async def async_runner():
         print("did setup")
         yield runner
     finally:
-        runner.shutdown()
+        await runner.shutdown()
 
 
 @pytest.mark.asyncio
@@ -76,7 +76,7 @@ async def test_prediction_runner_setup():
         assert isinstance(result.started_at, datetime)
         assert isinstance(result.completed_at, datetime)
     finally:
-        runner.shutdown()
+        await runner.shutdown()
 
 
 @pytest.mark.asyncio