From fefd2daf2bb94fd48e2d9701433a4213eb5d3a9f Mon Sep 17 00:00:00 2001
From: Mike Salvatore <mike.s.salvatore@gmail.com>
Date: Fri, 24 Jun 2022 19:21:48 -0400
Subject: [PATCH] Agent: Use Mapping instead of Dict

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

diff --git a/monkey/infection_monkey/master/automated_master.py b/monkey/infection_monkey/master/automated_master.py
index 003c9873c..9333554b7 100644
--- a/monkey/infection_monkey/master/automated_master.py
+++ b/monkey/infection_monkey/master/automated_master.py
@@ -1,7 +1,7 @@
 import logging
 import threading
 import time
-from typing import Any, Callable, Dict, Iterable, List, Optional
+from typing import Any, Callable, Iterable, List, Mapping, Optional
 
 from common.configuration import PluginConfiguration
 from common.utils import Timer
@@ -208,7 +208,7 @@ class AutomatedMaster(IMaster):
         self,
         plugins: Iterable[PluginConfiguration],
         callback: Callable[[Any], None],
-        custom_pba_options: Dict,
+        custom_pba_options: Mapping,
     ):
         self._run_plugins(plugins, "post-breach action", callback)