aboutsummaryrefslogtreecommitdiff
path: root/tests/test_nn_tensorflow_tflite_metrics.py
diff options
context:
space:
mode:
authorDmitrii Agibov <dmitrii.agibov@arm.com>2022-10-04 15:18:55 +0100
committerDmitrii Agibov <dmitrii.agibov@arm.com>2022-10-04 15:34:38 +0100
commit366ac7437383d3e9a99a971e61afbc5c466bbfcc (patch)
tree8fca425ad9f03817391bdc1ca9c6dfdc9dc44489 /tests/test_nn_tensorflow_tflite_metrics.py
parent2d6ec94a2a085e7fa4fc23c4c7dcda3206eec98c (diff)
downloadmlia-366ac7437383d3e9a99a971e61afbc5c466bbfcc.tar.gz
MLIA-673 Use inclusive language in the source code and filenames
- Update configuration for inclusive language linter - Fix reported issues Change-Id: If0f8b6e20c17d8ee1c6179c61040fc351437f036
Diffstat (limited to 'tests/test_nn_tensorflow_tflite_metrics.py')
-rw-r--r--tests/test_nn_tensorflow_tflite_metrics.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/test_nn_tensorflow_tflite_metrics.py b/tests/test_nn_tensorflow_tflite_metrics.py
index a5e7736..ca4ab55 100644
--- a/tests/test_nn_tensorflow_tflite_metrics.py
+++ b/tests/test_nn_tensorflow_tflite_metrics.py
@@ -17,8 +17,8 @@ from mlia.nn.tensorflow.tflite_metrics import ReportClusterMode
from mlia.nn.tensorflow.tflite_metrics import TFLiteMetrics
-def _dummy_keras_model() -> tf.keras.Model:
- # Create a dummy model
+def _sample_keras_model() -> tf.keras.Model:
+ # Create a sample model
keras_model = tf.keras.Sequential(
[
tf.keras.Input(shape=(8, 8, 3)),
@@ -40,7 +40,7 @@ def _sparse_binary_keras_model() -> tf.keras.Model:
value[...] = 1.0
return weights
- keras_model = _dummy_keras_model()
+ keras_model = _sample_keras_model()
# Assign weights to have 0.5 sparsity
for layer in keras_model.layers:
if not isinstance(layer, tf.keras.layers.Flatten):
@@ -74,7 +74,7 @@ class TestTFLiteMetrics:
@staticmethod
def test_sparsity(metrics: TFLiteMetrics) -> None:
"""Test sparsity."""
- # Create new instance with a dummy TFLite file
+ # Create new instance with a sample TFLite file
# Check sparsity calculation
sparsity_per_layer = metrics.sparsity_per_layer()
for name, sparsity in sparsity_per_layer.items():