From 13b7e470db6776c809d31ecb3ef9aad8e3352bef Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Thu, 7 Apr 2022 13:25:40 -0400 Subject: [PATCH] Agent: Set timeout to None for custom PBA --- monkey/infection_monkey/post_breach/custom_pba/custom_pba.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/monkey/infection_monkey/post_breach/custom_pba/custom_pba.py b/monkey/infection_monkey/post_breach/custom_pba/custom_pba.py index 453dfb6ed..ce7dd64f2 100644 --- a/monkey/infection_monkey/post_breach/custom_pba/custom_pba.py +++ b/monkey/infection_monkey/post_breach/custom_pba/custom_pba.py @@ -26,7 +26,9 @@ class CustomPBA(PBA): """ def __init__(self, telemetry_messenger: ITelemetryMessenger): - super(CustomPBA, self).__init__(telemetry_messenger, POST_BREACH_FILE_EXECUTION) + super(CustomPBA, self).__init__( + telemetry_messenger, POST_BREACH_FILE_EXECUTION, timeout=None + ) self.filename = "" def run(self, options: Dict) -> Iterable[PostBreachData]: