Merge branch 'main' into unpin-xgb

microsoft · Jan 4, 2024 · bfc0cc3 · bfc0cc3
2 parents a25cf45 + 1904db7
commit bfc0cc3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/hummingbird/ml/operator_converters/onnx/array_feature_extractor.py b/hummingbird/ml/operator_converters/onnx/array_feature_extractor.py
@@ -45,3 +45,4 @@ def convert_onnx_array_feature_extractor(operator, device, extra_config):
 
 
 register_converter("ONNXMLArrayFeatureExtractor", convert_onnx_array_feature_extractor)
+register_converter("ONNXMLGather", convert_onnx_array_feature_extractor)
diff --git a/hummingbird/ml/supported.py b/hummingbird/ml/supported.py
@@ -89,6 +89,7 @@
 Concat,
 Div,
 Imputer,
+Gather,
 LabelEncoder,
 Less,
 LinearClassifier,
@@ -357,6 +358,7 @@ def _build_onnxml_operator_list():
             "ArrayFeatureExtractor",
             "Binarizer",
             "FeatureVectorizer",
+            "Gather",
             "Imputer",
             "LabelEncoder",
             "OneHotEncoder",
Original file line number	Diff line number	Diff line change
Expand Up		@@ -45,3 +45,4 @@ def convert_onnx_array_feature_extractor(operator, device, extra_config):


		register_converter("ONNXMLArrayFeatureExtractor", convert_onnx_array_feature_extractor)
		register_converter("ONNXMLGather", convert_onnx_array_feature_extractor)