From 87942213a55d9a7c7c2baf38fd0608f5123a6ea4 Mon Sep 17 00:00:00 2001 From: Seaswimmer Date: Tue, 11 Jun 2024 03:17:38 -0400 Subject: [PATCH] fix(aurora): typehints --- aurora/models/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aurora/models/base.py b/aurora/models/base.py index 7f91e34..d3bc075 100644 --- a/aurora/models/base.py +++ b/aurora/models/base.py @@ -13,7 +13,7 @@ class AuroraBaseModel(BaseModel): def dump(self) -> dict: return self.model_dump(exclude={"bot"}) - def to_json(self, indent: int | None = None, file: Any | None = None, **kwargs): + def to_json(self, indent: int | None = None, file: Any | None = None, **kwargs) -> str: from ..utilities.json import dump, dumps # pylint: disable=cyclic-import return dump(self.dump(), file, indent=indent, **kwargs) if file else dumps(self.dump(), indent=indent, **kwargs) @@ -26,6 +26,6 @@ class AuroraGuildModel(AuroraBaseModel): def dump(self) -> dict: return self.model_dump(exclude={"bot", "guild_id", "guild"}) - def to_json(self, indent: int | None = None, file: Any | None = None, **kwargs): + def to_json(self, indent: int | None = None, file: Any | None = None, **kwargs) -> str: from ..utilities.json import dump, dumps # pylint: disable=cyclic-import return dump(self.dump(), file, indent=indent, **kwargs) if file else dumps(self.dump(), indent=indent, **kwargs)