Merge pull request #47 from severinsimmler/dev

v1.3.1

chaine/crf.pyx

@@ -377,6 +377,8 @@ cdef class Model:
         List[Dict[str, float]]
             Probability distributions over all labels for each token
         """
+        if not isinstance(sequence, list):
+            sequence = list(sequence)
         self._set_sequence(sequence)
         return [
             {label: self._marginal(label, index) for label in self.labels}

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "chaine"
-version = "1.3.0"
+version = "1.3.1"
 description = "Linear-chain conditional random fields for natural language processing"
 authors = ["Severin Simmler <[email protected]>"]
 readme = "README.md"

tests/test_crf.py

@@ -198,6 +198,15 @@ def test_model_predict(model, dataset):
     assert predicted == expected
 
 
+def test_model_predict_generator(model, dataset):
+    generator = (
+        (features for features in sequence) for sequence in dataset["sequences"]
+    )
+    predicted = model.predict(generator)
+    expected = dataset["labels"]
+    assert predicted == expected
+
+
 def test_model_predict_proba_single(model, dataset):
     for sequence in dataset["sequences"]:
         predicted = model.predict_proba_single(sequence)