Update bio_ner_predictor_test.py

asyml · Apr 20, 2022 · 03e056f · 03e056f
1 parent b562bc7
commit 03e056f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)