From 8da156ef7da84a9d11f5945b307d04a57b5341be Mon Sep 17 00:00:00 2001 From: Gabo Date: Mon, 12 Feb 2024 14:42:45 +0100 Subject: [PATCH] Fix test --- src/test/test_trainer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/test/test_trainer.py b/src/test/test_trainer.py index 3b8ad72..0d9cd14 100644 --- a/src/test/test_trainer.py +++ b/src/test/test_trainer.py @@ -29,5 +29,4 @@ def test_predict(self): self.assertEqual(TokenType.TITLE, tokens[0].token_type) self.assertEqual("Document Big Centered Title", tokens[0].content) self.assertEqual(TokenType.TEXT, tokens[1].token_type) - self.assertEqual(TokenType.TEXT, tokens[10].token_type) self.assertEqual("List Title", tokens[10].content)