diff --git a/src/apify/_actor.py b/src/apify/_actor.py index 4fd2989e..76f4475b 100644 --- a/src/apify/_actor.py +++ b/src/apify/_actor.py @@ -252,8 +252,12 @@ async def finalize() -> None: # Persist Actor state await self._save_actor_state() - await asyncio.wait_for(finalize(), self._cleanup_timeout.total_seconds()) - self._is_initialized = False + try: + await asyncio.wait_for(finalize(), self._cleanup_timeout.total_seconds()) + except TimeoutError: + self.log.exception('Actor cleanup timed out') + finally: + self._is_initialized = False if self._exit_process: sys.exit(self.exit_code)