diff --git a/monkey/infection_monkey/credential_collectors/i_credential_collector.py b/monkey/infection_monkey/credential_collectors/i_credential_collector.py index 846cd26ec..847cd929d 100644 --- a/monkey/infection_monkey/credential_collectors/i_credential_collector.py +++ b/monkey/infection_monkey/credential_collectors/i_credential_collector.py @@ -1,10 +1,10 @@ from abc import ABC, abstractmethod -from typing import List, Mapping, Union +from typing import Iterable, Mapping, Optional from .credentials import Credentials class ICredentialCollector(ABC): @abstractmethod - def collect_credentials(self, options: Union[Mapping, None]) -> List[Credentials]: + def collect_credentials(self, options: Optional[Mapping]) -> Iterable[Credentials]: pass