diff options
author | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2022-10-04 15:18:55 +0100 |
---|---|---|
committer | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2022-10-04 15:34:38 +0100 |
commit | 366ac7437383d3e9a99a971e61afbc5c466bbfcc (patch) | |
tree | 8fca425ad9f03817391bdc1ca9c6dfdc9dc44489 /tests/test_nn_tensorflow_tflite_metrics.py | |
parent | 2d6ec94a2a085e7fa4fc23c4c7dcda3206eec98c (diff) | |
download | mlia-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.py | 8 |
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(): |