diff options
Diffstat (limited to 'tests/test_target_ethos_u_data_collection.py')
-rw-r--r-- | tests/test_target_ethos_u_data_collection.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/test_target_ethos_u_data_collection.py b/tests/test_target_ethos_u_data_collection.py index 3868b95..e034884 100644 --- a/tests/test_target_ethos_u_data_collection.py +++ b/tests/test_target_ethos_u_data_collection.py @@ -162,7 +162,7 @@ def mock_performance_estimation( metrics = PerformanceMetrics( target, NPUCycles(1, 2, 3, 4, 5, 6), - MemoryUsage(1, 2, 3, 4, 5), + MemoryUsage(1, 2, 3, 4), LayerwisePerfInfo(layerwise_info=[]), ) monkeypatch.setattr( |