From 9f8463f707a1428862e9741335f5fbced2b6f565 Mon Sep 17 00:00:00 2001
From: Shreya Malviya <shreya.malviya@gmail.com>
Date: Wed, 30 Mar 2022 13:42:58 +0530
Subject: [PATCH] Agent: Modify PBA base class to accept options in its run
 method

---
 monkey/infection_monkey/post_breach/pba.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/monkey/infection_monkey/post_breach/pba.py b/monkey/infection_monkey/post_breach/pba.py
index 9222d5e1a..302830e74 100644
--- a/monkey/infection_monkey/post_breach/pba.py
+++ b/monkey/infection_monkey/post_breach/pba.py
@@ -1,6 +1,6 @@
 import logging
 import subprocess
-from typing import Iterable
+from typing import Dict, Iterable
 
 from common.utils.attack_utils import ScanStatus
 from infection_monkey.i_puppet.i_puppet import PostBreachData
@@ -30,7 +30,7 @@ class PBA:
         self.pba_data = []
         self.telemetry_messenger = telemetry_messenger
 
-    def run(self) -> Iterable[PostBreachData]:
+    def run(self, options: Dict) -> Iterable[PostBreachData]:
         """
         Runs post breach action command
         """