From 03e056ffe76a0a034149da375d0229b99070ebf3 Mon Sep 17 00:00:00 2001 From: Hector Date: Wed, 20 Apr 2022 11:53:56 -0400 Subject: [PATCH] Update bio_ner_predictor_test.py --- tests/wrappers/bio_ner_predictor_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/wrappers/bio_ner_predictor_test.py b/tests/wrappers/bio_ner_predictor_test.py index 5017278..a94cd34 100644 --- a/tests/wrappers/bio_ner_predictor_test.py +++ b/tests/wrappers/bio_ner_predictor_test.py @@ -102,7 +102,7 @@ def setUp(self): self.set_up_succeed = True def test_predict(self): - if set_up_succeed: + if self.set_up_succeed: for idx, data_pack in enumerate(self.pl.process_dataset(self.input_path)): ems = list(data_pack.get_data(EntityMention)) self.assertTrue(len(ems) > 0)