diff options
Diffstat (limited to 'tests/test_devices_ethosu_data_collection.py')
-rw-r--r-- | tests/test_devices_ethosu_data_collection.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/test_devices_ethosu_data_collection.py b/tests/test_devices_ethosu_data_collection.py index a4f37aa..84b9424 100644 --- a/tests/test_devices_ethosu_data_collection.py +++ b/tests/test_devices_ethosu_data_collection.py @@ -6,6 +6,7 @@ from unittest.mock import MagicMock import pytest +from mlia.backend.vela.compat import Operators from mlia.core.context import Context from mlia.core.data_collection import DataCollector from mlia.core.errors import FunctionalityNotSupportedError @@ -18,7 +19,6 @@ from mlia.devices.ethosu.performance import NPUCycles from mlia.devices.ethosu.performance import OptimizationPerformanceMetrics from mlia.devices.ethosu.performance import PerformanceMetrics from mlia.nn.tensorflow.optimizations.select import OptimizationSettings -from mlia.tools.vela_wrapper import Operators @pytest.mark.parametrize( |