fix type problems from mypy.

ScreenPyHQ · Feb 7, 2024 · 53b4711 · 53b4711
1 parent d2d864d
commit 53b4711
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/screenpy_playwright/abilities/browse_the_web_synchronously.py b/screenpy_playwright/abilities/browse_the_web_synchronously.py
@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from playwright.sync_api import Browser, Playwright, sync_playwright
+from playwright.sync_api import Browser, Page, Playwright, sync_playwright
 
 
 class BrowseTheWebSynchronously:
@@ -21,7 +21,8 @@ class BrowseTheWebSynchronously:
         )
     """
 
-    pages: list[type]
+    current_page: Page | None
+    pages: list[Page]
 
     @staticmethod
     def using(playwright: Playwright, browser: Browser) -> BrowseTheWebSynchronously:

diff --git a/screenpy_playwright/questions/text.py b/screenpy_playwright/questions/text.py
@@ -43,7 +43,7 @@ def describe(self) -> str:
         return f"The text from the {self.target}."
 
     @beat("{} examines the text of the {target}.")
-    def answered_by(self, the_actor: Actor) -> str:
+    def answered_by(self, the_actor: Actor) -> str | None:
         """Direct the Actor to read the text from the target.
 
         Returns: