diff --git a/grizzly/target/puppet_target.py b/grizzly/target/puppet_target.py index 3026159c..0a5a44a1 100644 --- a/grizzly/target/puppet_target.py +++ b/grizzly/target/puppet_target.py @@ -9,7 +9,7 @@ from platform import system from signal import SIGABRT from tempfile import TemporaryDirectory, mkdtemp -from typing import Any, Optional, cast +from typing import Any, cast from ffpuppet import BrowserTimeoutError, Debugger, FFPuppet, LaunchError, Reason from ffpuppet.helpers import certutil_available, certutil_find @@ -95,7 +95,7 @@ def __init__( **kwds: dict[str, Any], ) -> None: LOG.debug("ffpuppet version: %s", package_version("ffpuppet")) - certs = cast(Optional[CertificateBundle], kwds.pop("certs", None)) + certs = cast(CertificateBundle | None, kwds.pop("certs", None)) # only pass certs to FFPuppet if certutil is available # otherwise certs can't be used if certs and not certutil_available(certutil_find(binary)): @@ -126,7 +126,7 @@ def __init__( # create Puppet object self._puppet = FFPuppet( debugger=self._debugger, - headless=cast(Optional[str], kwds.pop("headless", None)), + headless=cast(str | None, kwds.pop("headless", None)), working_path=str(grz_tmp("target")), ) if kwds: @@ -259,7 +259,7 @@ def launch(self, location: str) -> None: memory_limit=self.memory_limit, prefs_js=self._prefs, extension=[self._extension] if self._extension else None, - env_mod=cast(dict[str, Optional[str]], env_mod), + env_mod=cast(dict[str, str | None], env_mod), cert_files=[self.certs.root] if self.certs else None, ) except LaunchError as exc: