diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/mlia/test_nn_tensorflow_tflite_metrics.py | 2 | ||||
| -rw-r--r-- | tests/mlia/utils/common.py | 2 |
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() |