Use type checking and future imports for parity

transprogrammer · Dec 30, 2023 · 177502e · 177502e
1 parent e64c268
commit 177502e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/bot/cogs/dev_tools.py b/bot/cogs/dev_tools.py
@@ -1,11 +1,15 @@
-from typing import Literal, Optional
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Literal, Optional
 
 import discord
 from cogs import EXTENSIONS
 from discord.ext import commands
 from discord.ext.commands import Context, Greedy
 from libs.utils import RoboContext, RoboView
-from rodhaj import Rodhaj
+
+if TYPE_CHECKING:
+    from bot.rodhaj import Rodhaj
 
 
 class MaybeView(RoboView):

diff --git a/bot/cogs/utilities.py b/bot/cogs/utilities.py
@@ -1,14 +1,19 @@
+from __future__ import annotations
+
 import datetime
 import itertools
 import platform
+from typing import TYPE_CHECKING
 
 import discord
 import psutil
 import pygit2
 from discord.ext import commands
 from discord.utils import format_dt
 from libs.utils import Embed, RoboContext, human_timedelta
-from rodhaj import Rodhaj
+
+if TYPE_CHECKING:
+    from bot.rodhaj import Rodhaj
 
 
 # A cog houses a category of commands