diff options
| author | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2022-11-28 13:36:03 +0000 |
|---|---|---|
| committer | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2022-12-06 17:05:37 +0000 |
| commit | 0241f96fce1bbfa11f21bfbfa161b51f3170a51b (patch) | |
| tree | f6ceae63670dd58f0b58de0ea55e039eddca3b50 /tests/test_nn_tensorflow_tflite_metrics.py | |
| parent | 6a88ee5315b4ce5b023370c1e55e48bf9f2b6f67 (diff) | |
| download | mlia-0241f96fce1bbfa11f21bfbfa161b51f3170a51b.tar.gz | |
MLIA-669 Upgrade dependencies with Vela 3.6
With Vela 3.6 we are able to remove the special treatment of aarch64
in our dependencies, i.e.
- upgrade Vela to version 3.6 that resolves a compatibility issue for
aarch64 in 3.4 and 3.5.
- upgrade to TensorFlow 2.10 which now supports aarch64 (therefore
making it obsolete to use 'tensorflow-aarch64').
Change-Id: I86508b667b5ccb55bfd11dcae9defc54e5ef74de
Diffstat (limited to 'tests/test_nn_tensorflow_tflite_metrics.py')
| -rw-r--r-- | tests/test_nn_tensorflow_tflite_metrics.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/test_nn_tensorflow_tflite_metrics.py b/tests/test_nn_tensorflow_tflite_metrics.py index 0e4c79c..e8d7c09 100644 --- a/tests/test_nn_tensorflow_tflite_metrics.py +++ b/tests/test_nn_tensorflow_tflite_metrics.py @@ -34,10 +34,10 @@ def _sample_keras_model() -> tf.keras.Model: def _sparse_binary_keras_model() -> tf.keras.Model: def get_sparse_weights(shape: list[int]) -> np.ndarray: weights = np.zeros(shape) - with np.nditer(weights, op_flags=["writeonly"]) as weight_iterator: - for idx, value in enumerate(weight_iterator): + with np.nditer(weights, op_flags=[["writeonly"]]) as weight_it: + for idx, value in enumerate(weight_it): if idx % 2 == 0: - value[...] = 1.0 + value[...] = 1.0 # type: ignore return weights keras_model = _sample_keras_model() |