diff --git a/backends/arm/operators/op_clamp.py b/backends/arm/operators/op_clamp.py index 67fff8b8a60..b18ed640b5f 100644 --- a/backends/arm/operators/op_clamp.py +++ b/backends/arm/operators/op_clamp.py @@ -63,7 +63,8 @@ def cast_type(value: Any) -> int | float: # Attempt to cast to float return float(value) - assert 2 <= len(node.args) <= 3 + if len(node.args) != 2 and len(node.args) != 3: + raise ValueError(f"Expected len(node.args) to be 2 or 3, got {node.args}") min_arg = dtype_min max_arg = dtype_max @@ -84,7 +85,10 @@ def define_node( inputs: List[TosaArg], output: TosaArg, ) -> None: - assert len(node.all_input_nodes) == 1 + if len(node.all_input_nodes) != 1: + raise ValueError( + f"Expected 1 input for {self.target}, got {len(node.all_input_nodes)}" + ) min_int8, max_int8 = self._get_min_max_arguments( node, @@ -122,7 +126,10 @@ def define_node( inputs: List[TosaArg], output: TosaArg, ) -> None: - assert len(node.all_input_nodes) == 1 + if len(node.all_input_nodes) != 1: + raise ValueError( + f"Expected 1 input for {self.target}, got {len(node.all_input_nodes)}" + ) if inputs[0].dtype == ts.DType.INT8: # Call the inherited define_node for handling integers