Update spacy/ml/tb_framework.pyx

Co-authored-by: Raphael Mitsch <r.mitsch@outlook.com>
This commit is contained in:
Adriane Boyd 2023-07-20 09:59:19 +02:00 committed by GitHub
parent 96f2e30c4b
commit 4f37e4031c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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")