Skip to content

Commit fed15ef

Browse files
Pass candidate identity into pooled Method 1 scoring
1 parent 6f3ddba commit fed15ef

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/nns/_stack_method1.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
CandidateId: TypeAlias = int | Literal["all"]
1212
Objective: TypeAlias = Literal["min", "max"]
1313
CandidateEvaluator: TypeAlias = Callable[
14-
[NDArray[np.float64], NDArray[np.float64]], tuple[float, float]
14+
[CandidateId, NDArray[np.float64], NDArray[np.float64]], tuple[float, float]
1515
]
1616

1717

@@ -119,7 +119,7 @@ def select_method1_candidate(
119119
complete[candidate] = False
120120
continue
121121

122-
score, threshold = evaluator(raw[valid], actual_values[valid])
122+
score, threshold = evaluator(candidate, raw[valid], actual_values[valid])
123123
score_value = float(score)
124124
threshold_value = float(threshold)
125125
if not math.isfinite(score_value):

0 commit comments

Comments
 (0)