commit 20e256d5f0be63864e59ad040fb60debefce0a28
author Sam Balana <sbalana@google.com> Mon Jun 05 22:19:22 2023 +0000
committer CQ Bot <fuchsia-internal-scoped@luci-project-accounts.iam.gserviceaccount.com> Mon Jun 05 22:19:22 2023 +0000
tree d4e7db14b83a9c87574e43449c21ce2ac817dceb
parent d3cc7b8f7beb33f9ccad8a7ca625728de5ecf7a9

Relax typing of pmf_support

Change-Id: I1f2e245d4fcabd1d87117b90e749d0cbaad7d66a
Reviewed-on: https://fuchsia-review.googlesource.com/c/antlion/+/865461
Commit-Queue: Auto-Submit <auto-submit@fuchsia-infra.iam.gserviceaccount.com>
Reviewed-by: Jonathan Chang <jnchang@google.com>
Fuchsia-Auto-Submit: Sam Balana <sbalana@google.com>

packages/antlion/controllers/access_point.py

diff --git a/packages/antlion/controllers/access_point.py b/packages/antlion/controllers/access_point.py
index bdb6923..477f6c8 100755
--- a/packages/antlion/controllers/access_point.py
+++ b/packages/antlion/controllers/access_point.py
@@ -17,7 +17,7 @@
 import ipaddress
 import time
 from dataclasses import dataclass
-from typing import Any, Dict, FrozenSet, List, Literal, Optional, Set, Tuple
+from typing import Any, Dict, FrozenSet, List, Optional, Set, Tuple
 
 from antlion import logger, utils
 from antlion.capabilities.ssh import SSHConfig, SSHProvider
@@ -751,7 +751,7 @@
     force_wmm: Optional[bool] = None,
     hidden: Optional[bool] = False,
     security: Optional[Security] = None,
-    pmf_support: Optional[Literal[1, 2, 3]] = None,
+    pmf_support: Optional[int] = None,
     additional_ap_parameters: Optional[Dict[str, Any]] = None,
     password: Optional[str] = None,
     n_capabilities: Optional[List[Any]] = None,

packages/antlion/controllers/ap_lib/hostapd_ap_preset.py

diff --git a/packages/antlion/controllers/ap_lib/hostapd_ap_preset.py b/packages/antlion/controllers/ap_lib/hostapd_ap_preset.py
index 1875c8b..cf3aab6 100644
--- a/packages/antlion/controllers/ap_lib/hostapd_ap_preset.py
+++ b/packages/antlion/controllers/ap_lib/hostapd_ap_preset.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Any, FrozenSet, List, Literal, Optional, TypeVar
+from typing import Any, FrozenSet, List, Optional, TypeVar
 
 from antlion import utils
 from antlion.controllers.ap_lib import hostapd_config, hostapd_constants, hostapd_utils
@@ -51,7 +51,7 @@
     mode: Optional[str] = None,
     frequency: Optional[int] = None,
     security: Optional[Security] = None,
-    pmf_support: Optional[Literal[1, 2, 3]] = None,
+    pmf_support: Optional[int] = None,
     ssid: Optional[str] = None,
     hidden: Optional[bool] = None,
     dtim_period: Optional[int] = None,

packages/antlion/controllers/ap_lib/hostapd_config.py

diff --git a/packages/antlion/controllers/ap_lib/hostapd_config.py b/packages/antlion/controllers/ap_lib/hostapd_config.py
index 51744d8..75c7229 100644
--- a/packages/antlion/controllers/ap_lib/hostapd_config.py
+++ b/packages/antlion/controllers/ap_lib/hostapd_config.py
@@ -14,7 +14,7 @@
 
 import collections
 import logging
-from typing import Any, Dict, FrozenSet, List, Literal, Optional, Union
+from typing import Any, Dict, FrozenSet, List, Optional, Union
 
 from antlion.controllers.ap_lib import hostapd_constants
 from antlion.controllers.ap_lib.hostapd_bss_settings import BssSettings
@@ -96,7 +96,7 @@
         security: Optional[Security] = None,
         bssid: Optional[str] = None,
         force_wmm: Optional[bool] = None,
-        pmf_support: Optional[Literal[1, 2, 3]] = None,
+        pmf_support: Optional[int] = None,
         obss_interval: Optional[int] = None,
         vht_channel_width: Optional[Any] = None,
         vht_center_channel: Optional[int] = None,