From 4f37e4031c227231d4a547d2e4fc306435a98a22 Mon Sep 17 00:00:00 2001 From: Adriane Boyd Date: Thu, 20 Jul 2023 09:59:19 +0200 Subject: [PATCH] Update spacy/ml/tb_framework.pyx Co-authored-by: Raphael Mitsch --- spacy/ml/tb_framework.pyx | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/spacy/ml/tb_framework.pyx b/spacy/ml/tb_framework.pyx index ed04045a6..a48c6b901 100644 --- a/spacy/ml/tb_framework.pyx +++ b/spacy/ml/tb_framework.pyx @@ -305,7 +305,8 @@ def _forward_fallback( seen_mask, is_train: bool, actions: Optional[List[Ints1d]] = None, - max_moves: int = 0): + max_moves: int = 0, +): nF = model.get_dim("nF") output = model.get_ref("output") hidden_b = model.get_param("hidden_b")