From e1b6d0460a92bce52b64314547b88c1dee32e271 Mon Sep 17 00:00:00 2001
From: colons <code@colons.co>
Date: Tue, 21 Dec 2021 10:14:28 +0000
Subject: [PATCH] Fix return type of find_elements* (#6645)

---
 .../selenium/webdriver/remote/webdriver.pyi   | 20 +++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/stubs/selenium/selenium/webdriver/remote/webdriver.pyi b/stubs/selenium/selenium/webdriver/remote/webdriver.pyi
index d2918e90b..2f117cb00 100644
--- a/stubs/selenium/selenium/webdriver/remote/webdriver.pyi
+++ b/stubs/selenium/selenium/webdriver/remote/webdriver.pyi
@@ -49,21 +49,21 @@ class WebDriver:
     @property
     def title(self) -> str: ...
     def find_element_by_id(self, id_) -> WebElement: ...
-    def find_elements_by_id(self, id_) -> WebElement: ...
+    def find_elements_by_id(self, id_) -> list[WebElement]: ...
     def find_element_by_xpath(self, xpath) -> WebElement: ...
-    def find_elements_by_xpath(self, xpath) -> WebElement: ...
+    def find_elements_by_xpath(self, xpath) -> list[WebElement]: ...
     def find_element_by_link_text(self, link_text) -> WebElement: ...
-    def find_elements_by_link_text(self, text) -> WebElement: ...
+    def find_elements_by_link_text(self, text) -> list[WebElement]: ...
     def find_element_by_partial_link_text(self, link_text) -> WebElement: ...
-    def find_elements_by_partial_link_text(self, link_text) -> WebElement: ...
+    def find_elements_by_partial_link_text(self, link_text) -> list[WebElement]: ...
     def find_element_by_name(self, name) -> WebElement: ...
-    def find_elements_by_name(self, name) -> WebElement: ...
+    def find_elements_by_name(self, name) -> list[WebElement]: ...
     def find_element_by_tag_name(self, name) -> WebElement: ...
-    def find_elements_by_tag_name(self, name) -> WebElement: ...
+    def find_elements_by_tag_name(self, name) -> list[WebElement]: ...
     def find_element_by_class_name(self, name) -> WebElement: ...
-    def find_elements_by_class_name(self, name) -> WebElement: ...
+    def find_elements_by_class_name(self, name) -> list[WebElement]: ...
     def find_element_by_css_selector(self, css_selector) -> WebElement: ...
-    def find_elements_by_css_selector(self, css_selector) -> WebElement: ...
+    def find_elements_by_css_selector(self, css_selector) -> list[WebElement]: ...
     def execute_script(self, script, *args): ...
     def execute_async_script(self, script, *args): ...
     @property
@@ -97,8 +97,8 @@ class WebDriver:
     def implicitly_wait(self, time_to_wait) -> None: ...
     def set_script_timeout(self, time_to_wait) -> None: ...
     def set_page_load_timeout(self, time_to_wait) -> None: ...
-    def find_element(self, by=..., value: Any | None = ...): ...
-    def find_elements(self, by=..., value: Any | None = ...): ...
+    def find_element(self, by=..., value: Any | None = ...) -> WebElement: ...
+    def find_elements(self, by=..., value: Any | None = ...) -> list[WebElement]: ...
     @property
     def desired_capabilities(self): ...
     def get_screenshot_as_file(self, filename) -> bool: ...