Update docstring in _verify_output_shape (#2372)

David Eriksson · facebook-github-bot · commit f3dd493f2343 · 2024-06-10T15:03:45.000-07:00
Summary: Pull Request resolved: #2372 See title Reviewed By: saitcakmak Differential Revision: D58376098 fbshipit-source-id: d5c274a1b726758d186f8a11bb89d38f481dbb73
diff --git a/botorch/utils/transforms.py b/botorch/utils/transforms.py
@@ -186,7 +186,7 @@ def _verify_output_shape(acqf: Any, X: Tensor, output: Tensor) -> bool:
         # acqf does not have model or acqf.model does not define `batch_shape`
         warnings.warn(
             "Output shape checks failed! Expected output shape to match t-batch shape"
-            f"of X, but got output with shape {output.shape} for X with shape"
+            f"of X, but got output with shape {output.shape} for X with shape "
             f"{X.shape}. Make sure that this is the intended behavior!",
             RuntimeWarning,
         )

Original file line number	Diff line number	Diff line change
`@@ -186,7 +186,7 @@ def _verify_output_shape(acqf: Any, X: Tensor, output: Tensor) -> bool:`
`186`	`186`	# acqf does not have model or acqf.model does not define `batch_shape`
`187`	`187`	`warnings.warn(`
`188`	`188`	`"Output shape checks failed! Expected output shape to match t-batch shape"`
`189`		`- f"of X, but got output with shape {output.shape} for X with shape"`
	`189`	`+ f"of X, but got output with shape {output.shape} for X with shape "`
`190`	`190`	`f"{X.shape}. Make sure that this is the intended behavior!",`
`191`	`191`	`RuntimeWarning,`
`192`	`192`	`)`