Fix

Signed-off-by: Tyler Gu <[email protected]>

acto/kubernetes_engine/base.py

@@ -1,23 +1,22 @@
 import subprocess
 import time
 from abc import ABC, abstractmethod
-from typing import Callable, List, Tuple
+from typing import Callable, Dict, List
 
 import kubernetes
 
 from acto.constant import CONST
 from acto.utils import get_thread_logger
 
 KubernetesEnginePostHookType = Callable[[kubernetes.client.ApiClient], None]
-KeyValuePairType = Tuple[str, str]
 
 
 class KubernetesEngine(ABC):
 
     @abstractmethod
     def __init__(self, acto_namespace: int,
                  posthooks: List[KubernetesEnginePostHookType] = None,
-                 feature_gates: List[KeyValuePairType] = None) -> None: ...
+                 feature_gates: Dict[str, str] = None) -> None: ...
     '''Constructor for KubernetesEngine
     
     Args:

acto/kubernetes_engine/kind.py

@@ -2,7 +2,7 @@
 import os
 import subprocess
 import time
-from typing import List
+from typing import Dict, List
 
 import kubernetes
 import yaml
@@ -17,7 +17,7 @@ class Kind(base.KubernetesEngine):
 
     def __init__(
             self, acto_namespace: int, posthooks: List[base.KubernetesEnginePostHookType] = None,
-            feature_gates: List[base.KeyValuePairType] = None):
+            feature_gates: Dict[str, str] = None):
         self._config_path = os.path.join(CONST.CLUSTER_CONFIG_FOLDER, f'KIND-{acto_namespace}.yaml')
         self._posthooks = posthooks
         self._feature_gates = feature_gates