Fix _parse.py

Signed-off-by: Pierre Bartet <[email protected]>
onnx · Jan 23, 2025 · f5be77c · f5be77c
1 parent 77310d5
commit f5be77c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/skl2onnx/_parse.py b/skl2onnx/_parse.py
@@ -362,6 +362,11 @@ def _parse_sklearn_column_transformer(scope, model, inputs, custom_parsers=None)
         elif isinstance(column_indices, (int, str)):
             column_indices = [column_indices]
         names = get_column_indices(column_indices, inputs, multiple=True)
+
+        # Skip transforms which apply to no columns at all
+        if len(names) == 0:
+            continue
+
         transform_inputs = []
         for onnx_var, onnx_is in names.items():
             tr_inputs = _fetch_input_slice(scope, [inputs[onnx_var]], onnx_is)