Fix WlanDriverRestartTest setup_class

Fix mypy errors introduced by a previous change
(https://fxrev.dev/922309) to refactor the method names of
wlan_policy_lib.

Change-Id: I57b8e430fb248327c872a2a8a66a9c1edab3ede1
Reviewed-on: https://fuchsia-review.googlesource.com/c/antlion/+/925155
Reviewed-by: Patrick Lu <patricklu@google.com>
Commit-Queue: Auto-Submit <auto-submit@fuchsia-infra.iam.gserviceaccount.com>
Fuchsia-Auto-Submit: Sam Balana <sbalana@google.com>
Reviewed-by: Zhiyi Chen <zhiyichen@google.com>
diff --git a/tests/wlan/functional/WlanDriverRestartTest.py b/tests/wlan/functional/WlanDriverRestartTest.py
index 2dfca22..0ea365a 100644
--- a/tests/wlan/functional/WlanDriverRestartTest.py
+++ b/tests/wlan/functional/WlanDriverRestartTest.py
@@ -44,20 +44,20 @@
         self.iterations = test_params.get("iterations", 10)
         self.fd = self.fuchsia_devices[0]
 
-        response = self.fd.ffx.run("driver list")
-        driver_list = str(response.stdout, "UTF-8")
+        driver_list_resp = self.fd.ffx.run("driver list")
+        driver_list = str(driver_list_resp.stdout, "UTF-8")
         if not driver_list.find("iwlwifi"):
             raise signals.TestSkip(
                 "No intel WiFi driver found on this device, skipping test"
             )
 
-        response = self.fd.sl4f.wlan_policy_lib.wlanCreateClientController()
+        response = self.fd.sl4f.wlan_policy_lib.create_client_controller()
         asserts.assert_is_none(
             response.get("error", None),
             f"Failed to create client controller. Err: {response['error']}",
         )
 
-        response = self.fd.sl4f.wlan_policy_lib.wlanStopClientConnections()
+        response = self.fd.sl4f.wlan_policy_lib.stop_client_connections()
         asserts.assert_is_none(
             response.get("error", None),
             f"Failed to stop client connections. Err: {response['error']}",