From e5b1ae295caeab3da7082feb986238c6883156a9 Mon Sep 17 00:00:00 2001 From: John Bauer Date: Thu, 28 Nov 2024 21:24:36 -0800 Subject: [PATCH] oops a test --- stanza/tests/mwt/test_character_classifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stanza/tests/mwt/test_character_classifier.py b/stanza/tests/mwt/test_character_classifier.py index 2ae0cc31b..1d8a699a2 100644 --- a/stanza/tests/mwt/test_character_classifier.py +++ b/stanza/tests/mwt/test_character_classifier.py @@ -81,7 +81,7 @@ def test_train(tmp_path): doc = CoNLL.conll2doc(input_str=ENG_DEV) dataloader = DataLoader(doc, 10, model.args, vocab=model.vocab, evaluation=True, expand_unk_vocab=True) preds = [] - for i, batch in enumerate(dataloader): + for i, batch in enumerate(dataloader.to_loader()): assert i == 0 # there should only be one batch preds += model.predict(batch, never_decode_unk=True, vocab=dataloader.vocab) assert len(preds) == 1