update unit test

bjascob · Jan 22, 2024 · bf81b58 · bf81b58
1 parent dd220c6
commit bf81b58
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/auto/ModelGenerateT5wtense.py b/tests/auto/ModelGenerateT5wtense.py
@@ -6,7 +6,7 @@
 import unittest
 import spacy        # requried or I get an error. From lazy loading? (spacy used to annotate graphs)
 import amrlib
-from   amrlib.models.generate_t5wtense.model_input_helper import ModelInputHelper
+from   amrlib.models.generate_xfm.model_input_helper import ModelInputHelper
 
 
 graph01 = '''
@@ -75,7 +75,7 @@ def testModelInputHelper(self):
         self.assertTrue(mih1.annotation_performed)
         mih2 = ModelInputHelper(graph02)                # string has annotations
         self.assertFalse(mih2.annotation_performed)
-        mih3 = ModelInputHelper(graph02, force_annotate=True)
+        mih3 = ModelInputHelper(graph02, reannotate=True)
         self.assertTrue(mih3.annotation_performed)
         self.assertEqual(mih1.get_tagged_oneline(), mih2.get_tagged_oneline())
         self.assertEqual(mih1.get_tagged_oneline(), mih3.get_tagged_oneline())