3 files changed, 8 insertions, 8 deletions
diff --git a/src/mlia/nn/tensorflow/config.py b/src/mlia/nn/tensorflow/config.py
index 0c3133a..d7d430f 100644
--- a/src/mlia/nn/tensorflow/config.py
+++ b/src/mlia/nn/tensorflow/config.py
@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
 # SPDX-License-Identifier: Apache-2.0
 """Model configuration."""
 from __future__ import annotations
@@ -110,8 +110,8 @@ def get_model(model: str | Path) -> ModelConfiguration:
     if is_saved_model(model):
         return TfModel(model)
 
-    raise Exception(
-        "The input model format is not supported"
+    raise ValueError(
+        "The input model format is not supported "
         "(supported formats: TensorFlow Lite, Keras, TensorFlow saved model)!"
     )
 
diff --git a/src/mlia/nn/tensorflow/optimizations/select.py b/src/mlia/nn/tensorflow/optimizations/select.py
index d4a8ea4..a78df12 100644
--- a/src/mlia/nn/tensorflow/optimizations/select.py
+++ b/src/mlia/nn/tensorflow/optimizations/select.py
@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
 # SPDX-License-Identifier: Apache-2.0
 """Module for optimization selection."""
 from __future__ import annotations
@@ -64,7 +64,7 @@ class OptimizationSettings(NamedTuple):
                 self.optimization_type, next_target, self.layers_to_optimize
             )
 
-        raise Exception(f"Unknown optimization type {self.optimization_type}")
+        raise ValueError(f"Optimization type {self.optimization_type} is unknown.")
 
 
 class MultiStageOptimizer(Optimizer):
diff --git a/src/mlia/nn/tensorflow/utils.py b/src/mlia/nn/tensorflow/utils.py
index 287e6ff..d688a63 100644
--- a/src/mlia/nn/tensorflow/utils.py
+++ b/src/mlia/nn/tensorflow/utils.py
@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
 # SPDX-FileCopyrightText: Copyright The TensorFlow Authors. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
 """Collection of useful functions for optimizations."""
@@ -22,7 +22,7 @@ def representative_dataset(
 ) -> Callable:
     """Sample dataset used for quantization."""
     if input_shape[0] != 1:
-        raise Exception("Only the input batch_size=1 is supported!")
+        raise ValueError("Only the input batch_size=1 is supported!")
 
     def dataset() -> Iterable:
         for _ in range(sample_count):
@@ -41,7 +41,7 @@ def get_tf_tensor_shape(model: str) -> list:
         default_signature = loaded.signatures[default_signature_key]
         inputs_tensor_info = default_signature.inputs
     except KeyError as err:
-        raise Exception(f"Signature '{default_signature_key}' not found") from err
+        raise KeyError(f"Signature '{default_signature_key}' not found.") from err
 
     return [
         dim