From ae9fed3c2b0ab97f9cfab41a5611df08143ae99a Mon Sep 17 00:00:00 2001 From: vakarisz Date: Tue, 15 Feb 2022 16:16:43 +0200 Subject: [PATCH] Agent: fixup typehints in ICredentialCollector --- .../credential_collectors/i_credential_collector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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