change .to_disk() to also persist component config (#22)

* fix issue with persistance to disk

* bump version -> 1.5.0

Author: talmago

spacy_crfsuite/about.py

@@ -1,5 +1,5 @@
 __title__ = "spacy_crfsuite"
-__version__ = "1.4.0"
+__version__ = "1.5.0"
 __summary__ = "spaCy pipeline component for CRF entity extraction"
 __author__ = "Tal Almagor"
 __email__ = "[email protected]"

spacy_crfsuite/crf_extractor.py

@@ -99,10 +99,15 @@ def from_disk(self, path: Union[Path, str] = "model.pkl") -> "CRFExtractor":
         Raises:
             IOError
         """
-        ent_tagger = joblib.load(path)
-        assert isinstance(ent_tagger, sklearn_crfsuite.estimator.CRF)
+        state_dict = joblib.load(path)
+
+        if isinstance(state_dict, sklearn_crfsuite.estimator.CRF):
+            self.ent_tagger = state_dict
+
+        elif isinstance(state_dict, dict):
+            for attr, attr_value in state_dict.items():
+                setattr(self, attr, attr_value)
 
-        self.ent_tagger = ent_tagger
         return self
 
     def to_disk(self, path: Union[Path, str] = "model.pkl") -> None:
@@ -115,7 +120,12 @@ def to_disk(self, path: Union[Path, str] = "model.pkl") -> None:
             RuntimeError - if entity tagger is not fitted for runtime.
         """
         self._check_runtime()
-        joblib.dump(self.ent_tagger, path)
+
+        state_dict = {
+            "component_config": self.component_config,
+            "ent_tagger": self.ent_tagger,
+        }
+        joblib.dump(state_dict, path)
 
     def use_dense_features(self) -> bool:
         """A predicate to test if dense features should be used, according