undo team template data wrapping

Author: ElectricalBoy

src/lpdb/async_session/async_session.py

@@ -6,7 +6,6 @@
 import aiohttp
 
 from ..session import AbstractLpdbSession, LpdbDataType, LpdbError
-from ..defs import TeamTemplate
 
 __all__ = ["AsyncLpdbSession"]
 
@@ -111,7 +110,7 @@ async def make_count_request(
     @override
     async def get_team_template(
         self, wiki: str, template: str, date: Optional[date] = None
-    ) -> Optional[TeamTemplate]:
+    ) -> Optional[dict[str, Any]]:
         params = {
             "wiki": wiki,
             "template": template,
@@ -124,16 +123,15 @@ async def get_team_template(
             parsed_response = await AsyncLpdbSession.__handle_response(response)
             if parsed_response[0] == None:
                 return None
-            return TeamTemplate(parsed_response[0])
+            return parsed_response[0]
 
     @override
     async def get_team_template_list(
         self, wiki: str, pagination: int = 1
-    ) -> list[TeamTemplate]:
+    ) -> list[dict[str, Any]]:
         async with self.__session.get(
             "teamtemplatelist",
             headers=self._get_header(),
             params={"wiki": wiki, "pagination": pagination},
         ) as response:
-            response = await AsyncLpdbSession.__handle_response(response)
-            return [TeamTemplate(data) for data in response]
+            return await AsyncLpdbSession.__handle_response(response)

src/lpdb/session.py

@@ -17,8 +17,6 @@
 
 import requests
 
-from .defs import TeamTemplate
-
 __all__ = ["LpdbError", "LpdbWarning", "LpdbSession"]
 
 type LpdbDataType = Literal[
@@ -181,7 +179,7 @@ def get_team_template(
         wiki: str,
         template: str,
         date: Optional[date] = None,
-    ) -> Optional[TeamTemplate]:
+    ) -> Optional[dict[str, Any]]:
         """
         Queries a team template from LPDB.
 
@@ -200,7 +198,7 @@ def get_team_template_list(
         self,
         wiki: str,
         pagination: int = 1,
-    ) -> list[TeamTemplate]:
+    ) -> list[dict[str, Any]]:
         """
         Queries a list of team template from LPDB.
 
@@ -317,7 +315,7 @@ def make_request(
     @override
     def get_team_template(
         self, wiki: str, template: str, date: Optional[date] = None
-    ) -> Optional[TeamTemplate]:
+    ) -> Optional[dict[str, Any]]:
         params = {
             "wiki": wiki,
             "template": template,
@@ -334,7 +332,7 @@ def get_team_template(
     @override
     def get_team_template_list(
         self, wiki: str, pagination: int = 1
-    ) -> list[TeamTemplate]:
+    ) -> list[dict[str, Any]]:
         lpdb_response = requests.get(
             AbstractLpdbSession.BASE_URL + "teamtemplatelist",
             headers=self._get_header(),