diff --git a/python/cog/server/runner.py b/python/cog/server/runner.py index 20f129937..73dc85ced 100644 --- a/python/cog/server/runner.py +++ b/python/cog/server/runner.py @@ -138,7 +138,7 @@ async def inner() -> SetupResult: self._terminating = self._mux.terminating = asyncio.Event() self._child.start() - self._ensure_event_reader() + self._start_event_reader() try: async for event in self._mux.read("SETUP", poll=0.1): @@ -343,7 +343,7 @@ def cancel(self, prediction_id: str) -> None: _read_events_task: "Optional[asyncio.Task[None]]" = None - def _ensure_event_reader(self) -> None: + def _start_event_reader(self) -> None: def handle_error(task: "asyncio.Task[None]") -> None: if task.cancelled(): return