From 408db2aad511be043b07126bd49ecdff04e75080 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= Date: Tue, 14 Nov 2023 18:59:41 +0100 Subject: [PATCH] Mark override fit funtcion --- src/PyClassifiers/PyClassifier.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/PyClassifiers/PyClassifier.h b/src/PyClassifiers/PyClassifier.h index ab98583..f55eb25 100644 --- a/src/PyClassifiers/PyClassifier.h +++ b/src/PyClassifiers/PyClassifier.h @@ -22,7 +22,7 @@ namespace pywrap { PyClassifier& fit(torch::Tensor& X, torch::Tensor& y, const std::vector& features, const std::string& className, std::map>& states) override; PyClassifier& fit(torch::Tensor& X, torch::Tensor& y); PyClassifier& fit(torch::Tensor& dataset, const std::vector& features, const std::string& className, std::map>& states) override { return *this; }; - PyClassifier& fit(torch::Tensor& dataset, const std::vector& features, const std::string& className, std::map>& states, const torch::Tensor& weights) { return *this; }; + PyClassifier& fit(torch::Tensor& dataset, const std::vector& features, const std::string& className, std::map>& states, const torch::Tensor& weights) override { return *this; }; torch::Tensor predict(torch::Tensor& X) override; std::vector predict(std::vector>& X) override { return std::vector(); }; float score(std::vector>& X, std::vector& y) override { return 0.0; };