Coverage for kwai/modules/training/teams/team_repository.py: 100%

1"""Module that defines an interface for a team repository.""" 

2from abc import ABC, abstractmethod 

3from typing import AsyncIterator 

4 

5from kwai.modules.training.teams.team import TeamEntity, TeamIdentifier 

6 

7 

8class TeamRepository(ABC): 

9 """Interface for a team repository.""" 

10 

11 @abstractmethod 

12 async def get_by_ids(self, *ids: TeamIdentifier) -> AsyncIterator[TeamEntity]: 

13 """Get all teams for the given ids. 

14 

15 Args: 

16 ids: A variable number of team ids. 

17 """ 

18 raise NotImplementedError