Improved type annotations

Author: WeirdCarrotMonster

yatq/worker/runner.py

@@ -12,7 +12,7 @@
 from yatq.exceptions import TaskRescheduleException
 from yatq.queue import Queue
 from yatq.worker.factory.base import BaseJobFactory
-from yatq.worker.job.base import BaseJob
+from yatq.worker.job.simple import BaseJob
 from yatq.worker.worker_settings import T_ExcInfo, WorkerSettings
 
 LOGGER = logging.getLogger("yatq.worker")

yatq/worker/worker_settings.py

@@ -4,7 +4,7 @@
 import aioredis
 
 from yatq.worker.factory.simple import SimpleJobFactory
-from yatq.worker.job.simple import SimpleJob
+from yatq.worker.job.base import BaseJob
 
 T_ExcInfo = Tuple[Type[BaseException], BaseException, TracebackType]
 
@@ -30,7 +30,7 @@ async def redis_client() -> aioredis.Redis:  # pragma: no cover
 
     @staticmethod
     async def on_task_process_exception(
-        job: SimpleJob,
+        job: BaseJob,
         exc_info: T_ExcInfo,
     ) -> None:  # pragma: no cover
         ...