Fix test

huridocs · Feb 12, 2024 · 8da156e · 8da156e
1 parent 7f46632
commit 8da156e
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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)