Fix xgboost and test errors

georgia-tech-db · Nov 17, 2023 · 8d2d5ef · 8d2d5ef
1 parent 3ca9ce1
commit 8d2d5ef
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/evadb/binder/statement_binder.py b/evadb/binder/statement_binder.py
@@ -92,7 +92,9 @@ def _bind_create_function_statement(self, node: CreateFunctionStatement):
                         outputs.append(column)
                     else:
                         inputs.append(column)
-            elif string_comparison_case_insensitive(node.function_type, "sklearn"):
+            elif string_comparison_case_insensitive(
+                node.function_type, "sklearn"
+            ) or string_comparison_case_insensitive(node.function_type, "XGBoost"):
                 assert (
                     "predict" in arg_map
                 ), f"Creating {node.function_type} functions expects 'predict' metadata."