Remove Fully Bayesian logic in low_rank

Summary: Remove the fully Bayesian specialty logic (previously necessary to deal with non-standard batch ranges).

Reviewed By: sdaulton

Differential Revision: D44634780

fbshipit-source-id: d589808345f4fe265ac5e30cf203fdecd49d960c

botorch/utils/low_rank.py

@@ -62,12 +62,9 @@ def _reshape_base_samples(
     mvn = posterior.distribution
     loc = mvn.loc
     peshape = posterior._extended_shape()
-    is_fully_b = int(isinstance(posterior, FullyBayesianPosterior))
     base_samples = base_samples.view(
-        sample_shape
-        + torch.Size([1 for _ in range(loc.ndim - 1 - is_fully_b)])
-        + peshape[-2 - is_fully_b :]
-    ).expand(sample_shape + loc.shape[: -1 - is_fully_b] + peshape[-2 - is_fully_b :])
+        sample_shape + torch.Size([1] * (loc.ndim - 1)) + peshape[-2:]
+    ).expand(sample_shape + loc.shape[:-1] + peshape[-2:])
     if posterior._is_mt:
         base_samples = _reshape_base_samples_non_interleaved(
             mvn=posterior.distribution,