Skip to content

Commit b8fdc3d

Browse files
Tonytikhomirovd
authored andcommitted
feature: faiss_mode selector added
# Conflicts: # hypex/matching.py
1 parent 5931021 commit b8fdc3d

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

hypex/matching.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -153,16 +153,16 @@ def _make_experiment(
153153
)
154154

155155
def __init__(
156-
self,
157-
group_match: bool = False,
158-
distance: Literal["mahalanobis", "l2"] = "mahalanobis",
159-
metric: Literal["atc", "att", "ate"] = "ate",
160-
bias_estimation: bool = True,
161-
quality_tests: (
162-
Literal["smd", "psi", "ks-test", "repeats", "t-test", "auto"]
163-
| list[Literal["smd", "psi", "ks-test", "repeats", "t-test", "auto"]]
164-
) = "auto",
165-
faiss_mode: Literal["base", "fast", "auto"] = "auto",
156+
self,
157+
group_match: bool = False,
158+
distance: Literal["mahalanobis", "l2"] = "mahalanobis",
159+
metric: Literal["atc", "att", "ate"] = "ate",
160+
bias_estimation: bool = True,
161+
quality_tests: (
162+
Literal["smd", "psi", "ks-test", "repeats", "t-test", "auto"]
163+
| list[Literal["smd", "psi", "ks-test", "repeats", "t-test", "auto"]]
164+
) = "auto",
165+
faiss_mode: Literal["base", "fast", "auto"] = "auto",
166166

167167
):
168168
super().__init__(

0 commit comments

Comments
 (0)