spancat: replace a cast by an assertion

danieldk · Sep 8, 2022 · 0bd5730 · 0bd5730
1 parent b828954
commit 0bd5730
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/spacy/pipeline/spancat.py b/spacy/pipeline/spancat.py
@@ -311,8 +311,9 @@ def set_annotations(self, docs: Iterable[Doc], activations: ActivationsT) -> Non
         labels = self.labels
 
         indices = activations["indices"]
+        scores = activations["scores"]
         assert isinstance(indices, Ragged)
-        scores = cast(Floats2d, activations["scores"])
+        assert isinstance(scores, Floats2d)
 
         offset = 0
         for i, doc in enumerate(docs):