Add missing test whether a valid kernel function has been provided.

SC-SGS · Nov 7, 2023 · f7f56bf · f7f56bf
1 parent db54557
commit f7f56bf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/bindings/Python/sklearn.cpp b/bindings/Python/sklearn.cpp
@@ -60,6 +60,9 @@ void parse_provided_params(svc &self, const py::kwargs &args) {
         std::stringstream ss{ args["kernel"].cast<std::string>() };
         plssvm::kernel_function_type kernel{};
         ss >> kernel;
+        if (ss.fail()) {
+            throw py::value_error{ fmt::format("'{}' is not in list", args["kernel"].cast<std::string>()) };
+        }
         self.svm_->set_params(plssvm::kernel_type = kernel);
     }
     if (args.contains("degree")) {