Fix incorrect return type from get_team.

cmu-sei · Feb 20, 2024 · 94e9f02 · 94e9f02
1 parent bc4b42e
commit 94e9f02
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/gamebrain/clients/gameboard.py b/gamebrain/clients/gameboard.py
@@ -117,12 +117,16 @@ class TeamDoesNotExist(Exception):
 
 async def get_team(team_id: str) -> TeamData | None:
     try:
-        return await _gameboard_get(f"teams/{team_id}")
+        data = await _gameboard_get(f"teams/{team_id}")
+        return TeamData(**data)
     except RequestFailure as e:
         if e.status_code == 400:
             raise TeamDoesNotExist
         else:
             return None
+    except ValidationError as e:
+        error(e)
+        return None
 
 
 async def get_teams(game_id: str):