diff --git a/monkey/infection_monkey/master/ip_scanner.py b/monkey/infection_monkey/master/ip_scanner.py index 1c273fa22..b54adfb4a 100644 --- a/monkey/infection_monkey/master/ip_scanner.py +++ b/monkey/infection_monkey/master/ip_scanner.py @@ -11,7 +11,9 @@ from .threading_utils import create_daemon_thread logger = logging.getLogger() -Callback = Callable[[str, PingScanData, Dict[int, PortScanData]], None] +IP = str +Port = int +Callback = Callable[[IP, PingScanData, Dict[Port, PortScanData]], None] class IPScanner: