aboutsummaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorRaul Farkas <raul.farkas@arm.com>2022-07-11 20:09:48 +0100
committerDiego Russo <diego.russo@arm.com>2022-07-22 10:59:29 +0000
commit625c280433fef3c9d1b64f58eab930ba0f89cd82 (patch)
treed7c1c9900abd76aa85515ad930572f9739a24dbe /tests
parent7a09acbd1bccc9a7f81d79ed57259a0d32aa6873 (diff)
downloadmlia-625c280433fef3c9d1b64f58eab930ba0f89cd82.tar.gz
MLIA-507 Upgrade Vela version
Upgrade Vela version from 3.3.0 to 3.4.0. - Adapt code to use new typing notation by replacing `numpy.array` with `numpy.ndarray` where necessary. Change-Id: I035e9564d448652aa09a52d79c71ef09663ea776
Diffstat (limited to 'tests')
-rw-r--r--tests/mlia/test_nn_tensorflow_tflite_metrics.py2
-rw-r--r--tests/mlia/utils/common.py2
2 files changed, 2 insertions, 2 deletions
diff --git a/tests/mlia/test_nn_tensorflow_tflite_metrics.py b/tests/mlia/test_nn_tensorflow_tflite_metrics.py
index 805f7d1..cf7aaeb 100644
--- a/tests/mlia/test_nn_tensorflow_tflite_metrics.py
+++ b/tests/mlia/test_nn_tensorflow_tflite_metrics.py
@@ -31,7 +31,7 @@ def _dummy_keras_model() -> tf.keras.Model:
def _sparse_binary_keras_model() -> tf.keras.Model:
- def get_sparse_weights(shape: List[int]) -> np.array:
+ 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):
diff --git a/tests/mlia/utils/common.py b/tests/mlia/utils/common.py
index 4313cde..932343e 100644
--- a/tests/mlia/utils/common.py
+++ b/tests/mlia/utils/common.py
@@ -7,7 +7,7 @@ import numpy as np
import tensorflow as tf
-def get_dataset() -> Tuple[np.array, np.array]:
+def get_dataset() -> Tuple[np.ndarray, np.ndarray]:
"""Return sample dataset."""
mnist = tf.keras.datasets.mnist
(x_train, y_train), _ = mnist.load_data()