From 6780c1f6d8c6c4a41670363dc0ea626c4cabd063 Mon Sep 17 00:00:00 2001 From: Jan Jirmasek Date: Fri, 4 Oct 2024 16:06:55 +0100 Subject: [PATCH] feedback from the code review --- coremltools/converters/mil/frontend/torch/ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/coremltools/converters/mil/frontend/torch/ops.py b/coremltools/converters/mil/frontend/torch/ops.py index a9f87bacd..aa7d0f23b 100644 --- a/coremltools/converters/mil/frontend/torch/ops.py +++ b/coremltools/converters/mil/frontend/torch/ops.py @@ -2384,7 +2384,7 @@ def _parse_positional_args(context, node) -> Tuple[Var]: # However, CoreML does not allow such empty tensor, so remove them now if np.any([is_tensor_empty(x) for x in xs]): filtered_xs = [x for x in xs if not is_tensor_empty(x)] - xs = filtered_xs if filtered_xs else [xs[0]] + xs = filtered_xs if len(filtered_xs) > 0 else [xs[0]] dim = inputs[1] if nargs > 1 else 0