diff options
Diffstat (limited to 'tests/test_nn_rewrite_core_graph_edit_record.py')
| -rw-r--r-- | tests/test_nn_rewrite_core_graph_edit_record.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tests/test_nn_rewrite_core_graph_edit_record.py b/tests/test_nn_rewrite_core_graph_edit_record.py index cd728af..41b9c50 100644 --- a/tests/test_nn_rewrite_core_graph_edit_record.py +++ b/tests/test_nn_rewrite_core_graph_edit_record.py @@ -3,15 +3,13 @@ """Tests for module mlia.nn.rewrite.graph_edit.record.""" from pathlib import Path -import pytest import tensorflow as tf from mlia.nn.rewrite.core.graph_edit.record import record_model from mlia.nn.rewrite.core.utils.numpy_tfrecord import numpytf_read -@pytest.mark.parametrize("batch_size", (None, 1, 2)) -def test_record_model( +def check_record_model( test_tflite_model: Path, tmp_path: Path, test_tfrecord: Path, |