diff options
Diffstat (limited to 'tests/mlia/test_devices_ethosu_performance.py')
-rw-r--r-- | tests/mlia/test_devices_ethosu_performance.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/mlia/test_devices_ethosu_performance.py b/tests/mlia/test_devices_ethosu_performance.py index e27efa0..b3e5298 100644 --- a/tests/mlia/test_devices_ethosu_performance.py +++ b/tests/mlia/test_devices_ethosu_performance.py @@ -23,6 +23,6 @@ def test_memory_usage_conversion() -> None: def mock_performance_estimation(monkeypatch: pytest.MonkeyPatch) -> None: """Mock performance estimation.""" monkeypatch.setattr( - "mlia.tools.aiet_wrapper.estimate_performance", + "mlia.backend.manager.estimate_performance", MagicMock(return_value=MagicMock()), ) |