Update botorch/utils/sampling.py

pytorch · Feb 24, 2025 · fd338fb · fd338fb
1 parent f637cc6
commit fd338fb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)