diff options
Diffstat (limited to 'tests/conftest.py')
-rw-r--r-- | tests/conftest.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tests/conftest.py b/tests/conftest.py index b1f32dc..feb2aa0 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -10,12 +10,12 @@ from typing import Generator import pytest import tensorflow as tf +from mlia.backend.vela.compiler import optimize_model from mlia.core.context import ExecutionContext from mlia.devices.ethosu.config import EthosUConfiguration from mlia.nn.tensorflow.utils import convert_to_tflite from mlia.nn.tensorflow.utils import save_keras_model from mlia.nn.tensorflow.utils import save_tflite_model -from mlia.tools.vela_wrapper import optimize_model @pytest.fixture(scope="session", name="test_resources_path") @@ -68,7 +68,9 @@ def test_resources(monkeypatch: pytest.MonkeyPatch, test_resources_path: Path) - """Return path to the test resources.""" return test_resources_path / "backends" - monkeypatch.setattr("mlia.backend.fs.get_backend_resources", get_test_resources) + monkeypatch.setattr( + "mlia.backend.executor.fs.get_backend_resources", get_test_resources + ) yield |