Skip to content

Commit 8d2d5ef

Browse files
committed
Fix xgboost and test errors
1 parent 3ca9ce1 commit 8d2d5ef

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

evadb/binder/statement_binder.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,9 @@ def _bind_create_function_statement(self, node: CreateFunctionStatement):
9292
outputs.append(column)
9393
else:
9494
inputs.append(column)
95-
elif string_comparison_case_insensitive(node.function_type, "sklearn"):
95+
elif string_comparison_case_insensitive(
96+
node.function_type, "sklearn"
97+
) or string_comparison_case_insensitive(node.function_type, "XGBoost"):
9698
assert (
9799
"predict" in arg_map
98100
), f"Creating {node.function_type} functions expects 'predict' metadata."

0 commit comments

Comments
 (0)