Fix get_decision_function (#645)

uxlfoundation · May 12, 2021 · edafa1d · edafa1d
1 parent 0bed689
commit edafa1d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/onedal/svm/backend/svm_py.cpp b/onedal/svm/backend/svm_py.cpp
@@ -286,7 +286,10 @@ PyObject *svm_infer<Task>::get_labels() {
 // attributes from infer_result
 template <typename Task>
 PyObject *svm_infer<Task>::get_decision_function() {
-    return convert_to_numpy(infer_result_.get_decision_function());
+    if constexpr (std::is_same_v<Task, svm::task::classification>) {
+        return convert_to_numpy(infer_result_.get_decision_function());
+    }
+    return nullptr;
 }
 
 template class ONEDAL_BACKEND_EXPORT svm_model<svm::task::classification>;