Skip to content

Commit dfc2449

Browse files
committed
Only pass batch_size parameter for interpoint constraints
1 parent ed752e8 commit dfc2449

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

baybe/recommenders/pure/bayesian/botorch.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -370,12 +370,18 @@ def _recommend_continuous_without_cardinality_constraints(
370370
raw_samples=self.n_raw_samples,
371371
fixed_features=fixed_parameters or None,
372372
equality_constraints=[
373-
c.to_botorch(subspace_continuous.parameters, batch_size=batch_size)
373+
c.to_botorch(
374+
subspace_continuous.parameters,
375+
batch_size=batch_size if c.is_interpoint else None,
376+
)
374377
for c in subspace_continuous.constraints_lin_eq
375378
]
376379
or None,
377380
inequality_constraints=[
378-
c.to_botorch(subspace_continuous.parameters, batch_size=batch_size)
381+
c.to_botorch(
382+
subspace_continuous.parameters,
383+
batch_size=batch_size if c.is_interpoint else None,
384+
)
379385
for c in subspace_continuous.constraints_lin_ineq
380386
]
381387
or None,

0 commit comments

Comments
 (0)