cbrkit 0.6.1__tar.gz → 0.6.2__tar.gz

{cbrkit-0.6.1 → cbrkit-0.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cbrkit
-Version: 0.6.1
+Version: 0.6.2
 Summary: Customizable Case-Based Reasoning (CBR) toolkit for Python with a built-in API and CLI.
 Home-page: https://wi2trier.github.io/cbrkit/
 License: MIT
@@ -41,9 +41,9 @@ Requires-Dist: sentence-transformers (>=2.2,<3.0) ; extra == "all" or extra == "
 Requires-Dist: spacy (>=3.7,<4.0) ; extra == "all" or extra == "all" or extra == "nlp"
 Requires-Dist: torch (>=2.1.1,<3.0.0) ; extra == "all" or extra == "transformers"
 Requires-Dist: transformers (>=4.35,<5.0) ; extra == "all" or extra == "transformers"
-Requires-Dist: typer[all] (>=0.9,<0.10) ; extra == "all" or extra == "cli"
+Requires-Dist: typer[all] (>=0.9,<1.0) ; extra == "all" or extra == "cli"
 Requires-Dist: uvicorn[standard] (>=0.24,<1.0) ; extra == "all" or extra == "api"
-Requires-Dist: xmltodict (>=0.13,<0.14)
+Requires-Dist: xmltodict (>=0.13,<1.0)
 Project-URL: Repository, https://github.com/wi2trier/cbrkit
 Description-Content-Type: text/markdown
 

{cbrkit-0.6.1 → cbrkit-0.6.2}/cbrkit/helpers.py

@@ -97,7 +97,12 @@ def sim2seq(
 
         return wrapped_func
 
-    return cast(SimSeqFunc[ValueType, SimType], func)
+    elif len(signature.parameters) == 1:
+        return cast(SimSeqFunc[ValueType, SimType], func)
+
+    raise TypeError(
+        f"Invalid signature for similarity function: {signature.parameters}"
+    )
 
 
 def sim2map(
@@ -107,7 +112,13 @@ def sim2map(
 ) -> SimMapFunc[KeyType, ValueType, SimType]:
     signature = inspect_signature(func)
 
-    if len(signature.parameters) == 2 and signature.parameters.keys() == {"x", "y"}:
+    if len(signature.parameters) == 2 and signature.parameters.keys() in (
+        {"x_map", "y"},
+        {"casebase", "query"},
+    ):
+        return cast(SimMapFunc[KeyType, ValueType, SimType], func)
+
+    elif len(signature.parameters) == 2:
         sim_pair_func = cast(SimPairFunc[ValueType, SimType], func)
 
         def wrapped_sim_pair_func(
@@ -131,7 +142,9 @@ def sim2map(
 
         return wrapped_sim_seq_func
 
-    return cast(SimMapFunc[KeyType, ValueType, SimType], func)
+    raise TypeError(
+        f"Invalid signature for similarity function: {signature.parameters}"
+    )
 
 
 def unpack_sim(sim: AnyFloat) -> float:

{cbrkit-0.6.1 → cbrkit-0.6.2}/cbrkit/retrieval.py

@@ -8,8 +8,8 @@ from cbrkit.typing import (
     AnySimFunc,
     Casebase,
     KeyType,
-    RetrieveFunc,
     SimMap,
+    SimMapFunc,
     SimType,
     ValueType,
 )
@@ -76,8 +76,8 @@ class Result(Generic[KeyType, ValueType, SimType]):
 def apply(
     casebase: Casebase[KeyType, ValueType],
     query: ValueType,
-    retrievers: RetrieveFunc[KeyType, ValueType, SimType]
-    | Sequence[RetrieveFunc[KeyType, ValueType, SimType]],
+    retrievers: SimMapFunc[KeyType, ValueType, SimType]
+    | Sequence[SimMapFunc[KeyType, ValueType, SimType]],
 ) -> Result[KeyType, ValueType, SimType]:
     """Applies a query to a Casebase using retriever functions.
 
@@ -135,7 +135,7 @@ def build(
     limit: int | None = None,
     min_similarity: float | None = None,
     max_similarity: float | None = None,
-) -> RetrieveFunc[KeyType, ValueType, SimType]:
+) -> SimMapFunc[KeyType, ValueType, SimType]:
     """Based on the similarity function this function creates a retriever function.
 
     The given limit will be applied after filtering for min/max similarity.
@@ -174,10 +174,10 @@ def build(
     sim_func = sim2map(similarity_func)
 
     def wrapped_func(
-        casebase: Casebase[KeyType, ValueType],
-        query: ValueType,
+        x_map: Casebase[KeyType, ValueType],
+        y: ValueType,
     ) -> SimMap[KeyType, SimType]:
-        similarities = sim_func(casebase, query)
+        similarities = sim_func(x_map, y)
         ranking = _similarities2ranking(similarities)
 
         if min_similarity is not None:
@@ -200,11 +200,11 @@ def build(
 
 def load(
     import_names: Sequence[str] | str,
-) -> list[RetrieveFunc[Any, Any, Any]]:
+) -> list[SimMapFunc[Any, Any, Any]]:
     if isinstance(import_names, str):
         import_names = [import_names]
 
-    retrievers: list[RetrieveFunc] = []
+    retrievers: list[SimMapFunc] = []
 
     for import_path in import_names:
         obj = load_python(import_path)
@@ -220,11 +220,11 @@ def load(
 
 def load_map(
     import_names: Collection[str] | str,
-) -> dict[str, RetrieveFunc[Any, Any, Any]]:
+) -> dict[str, SimMapFunc[Any, Any, Any]]:
     if isinstance(import_names, str):
         import_names = [import_names]
 
-    retrievers: dict[str, RetrieveFunc] = {}
+    retrievers: dict[str, SimMapFunc] = {}
 
     for import_path in import_names:
         obj = load_python(import_path)

{cbrkit-0.6.1 → cbrkit-0.6.2}/cbrkit/typing.py

@@ -28,9 +28,10 @@ SimSeq = Sequence[SimType]
 SimSeqOrMap = SimMap[KeyType, SimType] | SimSeq[SimType]
 
 
+# Parameter names must match so that the signature can be inspected, do not add `/` here!
 class SimMapFunc(Protocol[KeyType, ValueType_contra, SimType_cov]):
     def __call__(
-        self, x_map: Mapping[KeyType, ValueType_contra], y: ValueType_contra, /
+        self, x_map: Mapping[KeyType, ValueType_contra], y: ValueType_contra
     ) -> SimMap[KeyType, SimType_cov]:
         ...
 
@@ -42,9 +43,8 @@ class SimSeqFunc(Protocol[ValueType_contra, SimType_cov]):
         ...
 
 
-# Parameter names must match so that the signature can be inspected, do not add `/` here!
 class SimPairFunc(Protocol[ValueType_contra, SimType_cov]):
-    def __call__(self, x: ValueType_contra, y: ValueType_contra) -> SimType_cov:
+    def __call__(self, x: ValueType_contra, y: ValueType_contra, /) -> SimType_cov:
         ...
 
 
@@ -54,8 +54,6 @@ AnySimFunc = (
     | SimPairFunc[ValueType, SimType]
 )
 
-RetrieveFunc = SimMapFunc[KeyType, ValueType, SimType]
-
 
 class AggregatorFunc(Protocol[KeyType, SimType_contra]):
     def __call__(

{cbrkit-0.6.1 → cbrkit-0.6.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cbrkit"
-version = "0.6.1"
+version = "0.6.2"
 description = "Customizable Case-Based Reasoning (CBR) toolkit for Python with a built-in API and CLI."
 authors = ["Mirko Lenz <mirko@mirkolenz.com>"]
 license = "MIT"
@@ -52,13 +52,13 @@ sentence-transformers = { version = "^2.2", optional = true }
 spacy = { version = "^3.7", optional = true }
 torch = { version = "^2.1.1", optional = true }
 transformers = { version = "^4.35", optional = true }
-typer = { version = "^0.9", extras = ["all"], optional = true }
+typer = { version = ">=0.9, <1.0", extras = ["all"], optional = true }
 uvicorn = { version = ">=0.24, <1.0", optional = true, extras = ["standard"] }
-xmltodict = "^0.13"
+xmltodict = ">=0.13, <1.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.0.0"
-pytest-cov = "^4.1"
+pytest-cov = "^5.0.0"
 
 [tool.poetry.group.docs.dependencies]
 pdoc = "^14.4"