_tmux_session: Use updated namespace to catch typings

ansible · Feb 13, 2024 · d917ca5 · d917ca5
1 parent 9cf4677
commit d917ca5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/integration/_tmux_session.py b/tests/integration/_tmux_session.py
@@ -15,6 +15,7 @@
 from typing import TypedDict
 
 import libtmux
+import libtmux.exc
 import pytest
 
 from ._common import generate_test_log_dir
@@ -70,7 +71,7 @@ def __init__(
         self._pane_height = pane_height
         self._pane_width = pane_width
         self._pull_policy = pull_policy
-        self._session: libtmux.Session
+        self._session: libtmux.session.Session
         self._session_name = str(uuid.uuid4())
         self._setup_capture: str | list[str]
         self._setup_commands = setup_commands or []
@@ -114,7 +115,7 @@ def __enter__(self) -> TmuxSession:
         # pylint: disable=attribute-defined-outside-init
         # pylint: disable=too-many-locals
 
-        self._server = libtmux.Server()
+        self._server = libtmux.server.Server()
         self._build_tmux_session()
         self._window = self._session.new_window(self._session_name)
         self._pane = self._window.panes[0]