diff --git a/botorch/utils/sampling.py b/botorch/utils/sampling.py index adcb58fe2a..7066578b9d 100644 --- a/botorch/utils/sampling.py +++ b/botorch/utils/sampling.py @@ -1091,7 +1091,8 @@ def boltzmann_sample( succesively decreased by 'temp_decrease'. replacement: If True, samples are drawn with replacement, allowing duplicates. temp_decrease: The rate at which temperature decreases in case of inf weights. - Returns: + + Returns: A [batch_shape] x num_samples tensor of indices of sampled positions. """ norm_weights = standardize(function_values)