Presumably fix CI

thesadru · Sep 19, 2024 · 54dd961 · 54dd961
1 parent cac20dd
commit 54dd961
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 6 deletions.
diff --git a/genshin-dev/pytest-requirements.txt b/genshin-dev/pytest-requirements.txt
@@ -1,5 +1,5 @@
 pytest
-pytest-asyncio
+pytest-asyncio~=0.21.2
 pytest-cov
 pytest-dotenv
 

diff --git a/genshin/models/genshin/character.py b/genshin/models/genshin/character.py
@@ -71,6 +71,8 @@ def _get_db_char(
 ) -> constants.DBChar:
     """Get the appropriate DBChar object from specific fields."""
     if lang not in constants.CHARACTER_NAMES:
+        if id and name and icon and element and rarity:
+            return constants.DBChar(id or 0, _parse_icon(icon), name, element, rarity, guessed=True)
         raise Exception(
             f"Character names not loaded for {lang!r}. Please run `await genshin.utility.update_characters_any()`."
         )

diff --git a/genshin/models/genshin/chronicle/stats.py b/genshin/models/genshin/chronicle/stats.py
@@ -8,7 +8,8 @@
 from genshin.models import hoyolab
 from genshin.models.model import Aliased, APIModel
 
-from . import abyss, activities, characters
+from . import abyss, activities
+from . import characters as characters_module
 
 __all__ = [
     "AreaExploration",
@@ -159,7 +160,7 @@ class PartialGenshinUserStats(APIModel):
 
     info: hoyolab.UserInfo = Aliased("role")
     stats: Stats
-    characters: typing.Sequence["characters.PartialCharacter"] = Aliased("avatars")
+    characters: typing.Sequence[characters_module.PartialCharacter] = Aliased("avatars")
     explorations: typing.Sequence[Exploration] = Aliased("world_explorations")
     teapot: typing.Optional[Teapot] = Aliased("homes")
 
@@ -175,7 +176,7 @@ def __format_teapot(cls, v: typing.Any) -> typing.Optional[typing.Dict[str, typi
 class GenshinUserStats(PartialGenshinUserStats):
     """User stats with characters with equipment"""
 
-    characters: typing.Sequence["characters.Character"] = Aliased("avatars")
+    characters: typing.Sequence[characters_module.Character] = Aliased("avatars")
 
 
 class FullGenshinUserStats(GenshinUserStats):

diff --git a/genshin/models/honkai/chronicle/stats.py b/genshin/models/honkai/chronicle/stats.py
@@ -67,7 +67,7 @@ def __normalize_rank(cls, rank: typing.Optional[str]) -> typing.Optional[int]:
 
         return 69 - ord(rank)
 
-    model_config = pydantic.ConfigDict(frozen=False)
+    model_config: pydantic.ConfigDict = pydantic.ConfigDict(frozen=False)  # type: ignore
 
 
 # flake8: noqa: E222

diff --git a/genshin/models/model.py b/genshin/models/model.py
@@ -17,7 +17,7 @@
 class APIModel(pydantic.BaseModel):
     """Modified pydantic model."""
 
-    model_config = pydantic.ConfigDict(arbitrary_types_allowed=True)
+    model_config: pydantic.ConfigDict = pydantic.ConfigDict(arbitrary_types_allowed=True)  # type: ignore
 
 
 class Unique(abc.ABC):