fix: add set_context in base and remove it for samknn

adaptive-machine-learning · May 14, 2024 · 4baad0f · 4baad0f
1 parent 50f55ce
commit 4baad0f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/capymoa/base.py b/src/capymoa/base.py
@@ -152,6 +152,8 @@ def __init__(self, moa_learner, schema=None, CLI=None, random_seed=1):
 
         self.moa_learner.prepareForUse()
         self.moa_learner.resetLearningImpl()
+        self.moa_learner.setModelContext(schema.get_moa_header())
+
 
     def __str__(self):
         # Removes the package information from the name of the learner.

diff --git a/src/capymoa/classifier/_samknn.py b/src/capymoa/classifier/_samknn.py
@@ -76,6 +76,3 @@ def __init__(
             CLI=config_str,
             moa_learner=self.moa_learner,
         )
-
-        # setModelContext needs to be called for SAMkNN to initialise the short and long term memory
-        self.moa_learner.setModelContext(schema.get_moa_header())