diff options
Diffstat (limited to 'tests/test_backend_manager.py')
-rw-r--r-- | tests/test_backend_manager.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tests/test_backend_manager.py b/tests/test_backend_manager.py index a1e9198..c22b4c1 100644 --- a/tests/test_backend_manager.py +++ b/tests/test_backend_manager.py @@ -15,7 +15,6 @@ import pytest from mlia.backend.application import get_application from mlia.backend.execution import ExecutionContext -from mlia.backend.execution import run_application from mlia.backend.manager import BackendRunner from mlia.backend.manager import DeviceInfo from mlia.backend.manager import estimate_performance @@ -377,7 +376,7 @@ class TestBackendRunner: expected_command: list[str], ) -> None: """Test method run_application with local systems.""" - run_app = MagicMock(wraps=run_application) + run_app = MagicMock() monkeypatch.setattr("mlia.backend.manager.run_application", run_app) backend_runner = BackendRunner() |