✅ Updating tests

saber/tests/test_bert_for_ner.py

@@ -321,16 +321,16 @@ def test_predict_mt(self, bert_for_ner_specify_mt):
 
     def test_prepare_optimizers(self, bert_for_ner_specify):
         """Asserts that the returned optimizer object is as expected after call to
-        `BertForNER.prepare_optimizers()`.
+        `BertForNER._prepare_optimizers()`.
         """
-        actual = bert_for_ner_specify.prepare_optimizers()
+        actual = bert_for_ner_specify._prepare_optimizers()
 
         assert all(isinstance(opt, AdamW) for opt in actual)
 
     def test_prepare_optimizers_mt(self, bert_for_ner_specify_mt):
         """Asserts that the returned optimizer object is as expected after call to
-        `BertForNER.prepare_optimizers()` for a multi-task model.
+        `BertForNER._prepare_optimizers()` for a multi-task model.
         """
-        actual = bert_for_ner_specify_mt.prepare_optimizers()
+        actual = bert_for_ner_specify_mt._prepare_optimizers()
 
         assert all(isinstance(opt, AdamW) for opt in actual)

saber/tests/test_bert_for_ner_and_re.py

@@ -212,14 +212,14 @@ def test_predict_mt(self):
 
     def test_prepare_optimizers(self, bert_for_ner_and_re_specify):
         """Asserts that the returned optimizer object is as expected after call to
-        `BertForNERAndRE.prepare_optimizers()`.
+        `BertForNERAndRE._prepare_optimizers()`.
         """
-        actual = bert_for_ner_and_re_specify.prepare_optimizers()
+        actual = bert_for_ner_and_re_specify._prepare_optimizers()
 
         assert all(isinstance(opt, AdamW) for opt in actual)
 
     def test_prepare_optimizers_mt(self):
         """Asserts that the returned optimizer object is as expected after call to
-        `BertForNERAndRE.prepare_optimizers()` for a multi-task model.
+        `BertForNERAndRE._prepare_optimizers()` for a multi-task model.
         """
         pass

saber/tests/test_bert_utils.py

@@ -212,7 +212,7 @@ def test_get_dataloader_for_bert_conll2004(self, conll2004datasetreader_load, be
 
     def test_get_bert_optimizer(self, bert_for_ner_specify):
         """Asserts that the returned optimizer object is as expected after call to
-        `BertForNER.prepare_optimizers()` for a single-task model.
+        `bert_utils.get_bert_optimizer()` for a single-task model.
         """
         model, config = bert_for_ner_specify.model, bert_for_ner_specify.config