From 0241f96fce1bbfa11f21bfbfa161b51f3170a51b Mon Sep 17 00:00:00 2001 From: Benjamin Klimczak Date: Mon, 28 Nov 2022 13:36:03 +0000 Subject: 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 --- tests/test_nn_tensorflow_tflite_metrics.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'tests/test_nn_tensorflow_tflite_metrics.py') 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() -- cgit v1.2.1