diff --git a/screenpy_playwright/__init__.py b/screenpy_playwright/__init__.py index f19a240..f8a0d13 100644 --- a/screenpy_playwright/__init__.py +++ b/screenpy_playwright/__init__.py @@ -12,14 +12,14 @@ :license: MIT, see LICENSE for more details. """ -from . import abilities, actions, protocols, questions +from . import abilities, actions, questions from .abilities import * # noqa: F403 from .actions import * # noqa: F403 from .exceptions import TargetingError -from .protocols import * # noqa: F403 +from .protocols import PageObject from .questions import * # noqa: F403 from .target import Target -__all__ = ["Target", "TargetingError"] +__all__ = ["Target", "TargetingError", "PageObject"] -__all__ += abilities.__all__ + actions.__all__ + questions.__all__ + protocols.__all__ +__all__ += abilities.__all__ + actions.__all__ + questions.__all__ diff --git a/screenpy_playwright/protocols.py b/screenpy_playwright/protocols.py index 0390901..f42a3d8 100644 --- a/screenpy_playwright/protocols.py +++ b/screenpy_playwright/protocols.py @@ -7,6 +7,3 @@ class PageObject(Protocol): """PageObjects have a URL.""" url: str - - -__all__ = ["PageObject"]