diff --git a/jarvis_tasks/tasks/ban.py b/jarvis_tasks/tasks/ban.py index 3513c29..8e2c65d 100644 --- a/jarvis_tasks/tasks/ban.py +++ b/jarvis_tasks/tasks/ban.py @@ -1,16 +1,13 @@ """JARVIS ban tasks.""" import asyncio from datetime import datetime, timedelta -from typing import TYPE_CHECKING +from dis_snek import Snake from jarvis_core.db import q from jarvis_core.db.models import Ban, Unban -if TYPE_CHECKING: - from dis_snek import Snake - -async def unban(bot: "Snake") -> None: +async def unban(bot: Snake) -> None: """ Unban users when ban expires. diff --git a/jarvis_tasks/tasks/reminder.py b/jarvis_tasks/tasks/reminder.py index f51b054..faa1351 100644 --- a/jarvis_tasks/tasks/reminder.py +++ b/jarvis_tasks/tasks/reminder.py @@ -1,17 +1,14 @@ """JARVIS reminders.""" import asyncio from datetime import datetime, timedelta -from typing import TYPE_CHECKING +from dis_snek import Snake from jarvis_core.db import q from jarvis_core.db.models import Reminder from jarvis_core.util import build_embed -if TYPE_CHECKING: - from dis_snek import Snake - -async def remind(bot: "Snake") -> None: +async def remind(bot: Snake) -> None: """ Run reminders in the background. diff --git a/jarvis_tasks/tasks/twitter.py b/jarvis_tasks/tasks/twitter.py index 850db97..004b8d5 100644 --- a/jarvis_tasks/tasks/twitter.py +++ b/jarvis_tasks/tasks/twitter.py @@ -1,17 +1,14 @@ """JARVIS Twitter sync.""" import asyncio from datetime import datetime, timedelta -from typing import TYPE_CHECKING import tweepy +from dis_snek import Snake from jarvis_core.db import q from jarvis_core.db.models import TwitterAccount, TwitterFollow from jarvis_tasks.config import TaskConfig -if TYPE_CHECKING: - from dis_snek import Snake - config = TaskConfig.from_yaml()