Agent: Modify remaining PBAs to yield PostBreachData

monkey/infection_monkey/post_breach/actions/schedule_jobs.py

@@ -23,4 +23,4 @@ class ScheduleJobs(PBA):
     def run(self):
         post_breach_data = super(ScheduleJobs, self).run()
         remove_scheduled_jobs()
-        return post_breach_data
+        yield post_breach_data

monkey/infection_monkey/post_breach/actions/use_signed_scripts.py

@@ -24,7 +24,7 @@ class SignedScriptProxyExecution(PBA):
                 original_comspec = subprocess.check_output(  # noqa: DUO116
                     "if defined COMSPEC echo %COMSPEC%", shell=True
                 ).decode()
-            return super().run()
+            yield super().run()
         except Exception as e:
             logger.warning(
                 f"An exception occurred on running PBA "