aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--bot/seasons/__init__.py10
1 files changed, 7 insertions, 3 deletions
diff --git a/bot/seasons/__init__.py b/bot/seasons/__init__.py
index 1569f29a..1292485c 100644
--- a/bot/seasons/__init__.py
+++ b/bot/seasons/__init__.py
@@ -29,10 +29,14 @@ def get_extensions() -> List[str]:
extensions = []
for package in pkgutil.iter_modules([base_path]):
- package_path = base_path.joinpath(package.name)
- for module in pkgutil.iter_modules([package_path]):
- extensions.append(f"bot.seasons.{package.name}.{module.name}")
+ if package.ispkg:
+ package_path = base_path.joinpath(package.name)
+
+ for module in pkgutil.iter_modules([package_path]):
+ extensions.append(f"bot.seasons.{package.name}.{module.name}")
+ else:
+ extensions.append(f"bot.seasons.{package.name}")
return extensions