From bf81b587ea7b0bbf1300d62f00056940b98b3d1d Mon Sep 17 00:00:00 2001 From: Brad Jascob <22728060+bjascob@users.noreply.github.com> Date: Sun, 21 Jan 2024 17:47:10 -0700 Subject: [PATCH] update unit test --- tests/auto/ModelGenerateT5wtense.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/auto/ModelGenerateT5wtense.py b/tests/auto/ModelGenerateT5wtense.py index 2c811a3..e2a21cc 100644 --- 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())