diff --git a/pysentence_similarity/__init__.py b/pysentence_similarity/__init__.py index 3862a3d..56ce6f5 100644 --- a/pysentence_similarity/__init__.py +++ b/pysentence_similarity/__init__.py @@ -2,6 +2,5 @@ from ._model import Model from ._splitter import Splitter from ._storage import Storage -from ._utils import compute_score -__all__ = ["Model", "Splitter", "Storage", "compute_score"] +__all__ = ["Model", "Splitter", "Storage"] diff --git a/pysentence_similarity/_utils.py b/pysentence_similarity/utils.py similarity index 100% rename from pysentence_similarity/_utils.py rename to pysentence_similarity/utils.py diff --git a/tests/test_utils.py b/tests/test_utils.py index ef02ffb..21366ac 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -2,7 +2,8 @@ import unittest import numpy as np -from pysentence_similarity import Model, compute_score +from pysentence_similarity import Model +from pysentence_similarity.utils import compute_score class TestModel(unittest.TestCase):