From cb548ff2210fc354750b3b64f27b386bebf5f571 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?=
 <rmontanana@gmail.com>
Date: Fri, 9 Feb 2024 11:26:39 +0100
Subject: [PATCH] Add getNotes method

---
 src/PyClassifiers/PyClassifier.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/PyClassifiers/PyClassifier.h b/src/PyClassifiers/PyClassifier.h
index 7260d2e..0686546 100644
--- a/src/PyClassifiers/PyClassifier.h
+++ b/src/PyClassifiers/PyClassifier.h
@@ -40,10 +40,12 @@ namespace pywrap {
         bayesnet::status_t getStatus() const override { return bayesnet::NORMAL; };
         std::vector<std::string> topological_order() override { return std::vector<std::string>(); }
         void dump_cpt() const override {};
+        std::vector<std::string> getNotes() const override { return notes; };
         void setHyperparameters(const nlohmann::json& hyperparameters) override;
     protected:
         nlohmann::json hyperparameters;
         void trainModel(const torch::Tensor& weights) override {};
+        std::vector<std::string> notes;
     private:
         PyWrap* pyWrap;
         std::string module;