diff --git a/dvc/repo/live.py b/dvc/repo/live.py index 8d5a79cdb..abd803df3 100644 --- a/dvc/repo/live.py +++ b/dvc/repo/live.py @@ -1,5 +1,4 @@ import logging -import webbrowser from pathlib import Path from platform import uname from typing import TYPE_CHECKING, List, Optional @@ -14,7 +13,12 @@ if TYPE_CHECKING: from dvc.output import Output from dvc.repo import Repo -webbrowser_open = once_per_args(webbrowser.open) + +@once_per_args +def webbrowser_open(url: str) -> int: + import webbrowser + + return webbrowser.open(url) def create_summary(out): diff --git a/tests/conftest.py b/tests/conftest.py index 259903854..c08c370bc 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -180,7 +180,6 @@ def custom_template(tmp_dir, dvc): scriptify_fixture = pytest.fixture(lambda: scriptify, name="scriptify") -@pytest.fixture(scope="session", autouse=True) -def webbrowser_open(session_mocker): - session_mocker.patch("dvc.repo.live.webbrowser_open") - session_mocker.patch("webbrowser.open") +@pytest.fixture(autouse=True) +def mocked_webbrowser_open(mocker): + mocker.patch("webbrowser.open")