Merge pull request #164 from sarugaku:fix/criterion

fix: export criterion in resolvers to keep compatibility
sarugaku · Aug 1, 2024 · ff700ea · ff700ea
2 parents d4a1da9 + d8be45d
commit ff700ea
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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():