diff --git a/src/resolvelib/reporters.py b/src/resolvelib/reporters.py index 2ce4626..26c9f6e 100644 --- a/src/resolvelib/reporters.py +++ b/src/resolvelib/reporters.py @@ -5,7 +5,7 @@ from .structs import CT, KT, RT, RequirementInformation, State if TYPE_CHECKING: - from .resolvers.criterion import Criterion + from .resolvers import Criterion class BaseReporter(Generic[RT, CT, KT]): diff --git a/src/resolvelib/resolvers/__init__.py b/src/resolvelib/resolvers/__init__.py index 263d4ff..7b2c5d5 100644 --- a/src/resolvelib/resolvers/__init__.py +++ b/src/resolvelib/resolvers/__init__.py @@ -1,5 +1,6 @@ from ..structs import RequirementInformation from .abstract import AbstractResolver, Result +from .criterion import Criterion from .exceptions import ( InconsistentCandidate, RequirementsConflicted, @@ -22,4 +23,5 @@ "RequirementInformation", "ResolverException", "Result", + "Criterion", ] diff --git a/tests/test_resolvers.py b/tests/test_resolvers.py index af29eb5..0270b18 100644 --- a/tests/test_resolvers.py +++ b/tests/test_resolvers.py @@ -18,8 +18,11 @@ if TYPE_CHECKING: from typing import Iterable, Mapping - from resolvelib.resolvers import RequirementInformation, RequirementsConflicted - from resolvelib.resolvers.criterion import Criterion + from resolvelib.resolvers import ( + Criterion, + RequirementInformation, + RequirementsConflicted, + ) def test_candidate_inconsistent_error():