diff options
Diffstat (limited to 'tests/mlia/utils/common.py')
-rw-r--r-- | tests/mlia/utils/common.py | 2 |
1 files changed, 1 insertions, 1 deletions
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() |