Fix caching

The cache was not restoring overwritten parameters
gmrukwa · Feb 20, 2021 · 0b9414f · 0b9414f
1 parent 6d85b20
commit 0b9414f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/divik/core/_cache.py b/divik/core/_cache.py
@@ -35,6 +35,7 @@ def fit(self, X, y=None):
         memory = joblib.Memory(location=cache_path())
         cached = memory.cache(_fit)
         output = cached(self, X, y)
+        self.set_params(**output.get_params())
         computed_fields = [f for f in dir(output) if _is_computed(f)]
         for field in computed_fields:
             setattr(self, field, getattr(output, field))

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "divik"
-version = "3.0.8"
+version = "3.0.9"
 description = "Divisive iK-means algorithm implementation"
 authors = ["Grzegorz Mrukwa <g.mrukwa@gmail.com>"]
 license = "Apache-2.0"