model: transformers: example: Fix accuracy assertion

Fixes: #497

Signed-off-by: John Andersen <[email protected]>

Author: pdxjohnny

model/transformers/examples/ner/test_ner_model.py

@@ -31,7 +31,8 @@ def python_test(self, filename):
         stdout = subprocess.check_output([sys.executable, filepath])
         lines = stdout.decode().split("\n")
         # Check the Accuracy
-        self.assertIn("Accuracy: 0.0", lines)
+        if not list(filter(lambda line: line.startswith("Accuracy: "), lines)):
+            raise AssertionError(f"Accuracy not found in: {lines}")
         # Check the predicted tag
         for line in lines:
             try: