Skip to content

Commit f8b3f17

Browse files
committed
Fix mypy error
1 parent 23b0309 commit f8b3f17

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

baybe/searchspace/discrete.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -423,6 +423,7 @@ def from_simplex(
423423
# discarded, because it is already clear that the total sum will be exceeded
424424
# once all joins are completed. Analogously, nonzero cardinality bounds are
425425
# checked at each step.
426+
arr: np.ndarray
426427
for i, (
427428
param,
428429
min_sum_to_go,
@@ -443,7 +444,7 @@ def from_simplex(
443444
else:
444445
n_old = arr.shape[0]
445446
n_new = len(values)
446-
arr = np.column_stack( # type: ignore[assignment] # shape widens
447+
arr = np.column_stack(
447448
[
448449
np.repeat(arr, n_new, axis=0),
449450
np.tile(values, n_old),

0 commit comments

Comments
 (0)