We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 23b0309 commit f8b3f17Copy full SHA for f8b3f17
1 file changed
baybe/searchspace/discrete.py
@@ -423,6 +423,7 @@ def from_simplex(
423
# discarded, because it is already clear that the total sum will be exceeded
424
# once all joins are completed. Analogously, nonzero cardinality bounds are
425
# checked at each step.
426
+ arr: np.ndarray
427
for i, (
428
param,
429
min_sum_to_go,
@@ -443,7 +444,7 @@ def from_simplex(
443
444
else:
445
n_old = arr.shape[0]
446
n_new = len(values)
- arr = np.column_stack( # type: ignore[assignment] # shape widens
447
+ arr = np.column_stack(
448
[
449
np.repeat(arr, n_new, axis=0),
450
np.tile(values, n_old),
0 commit comments