Add imports

stanfordnlp · Jan 22, 2024 · 7cc5f2d · 7cc5f2d
1 parent cf2b523
commit 7cc5f2d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/unit_tests/ModelUtilsTestCase.py b/tests/unit_tests/ModelUtilsTestCase.py
@@ -1,5 +1,6 @@
 import unittest
 from ..utils import *
+from pyvene.models.modeling_utils import *
 
 class ModelUtilsTestCase(unittest.TestCase):
     @classmethod