Add _supports_batched_models attribute to models that don't support batching (meta-pytorch#3239)

Summary:
Pull Request resolved: meta-pytorch#3239

This is needed update `use_model_list` in the stacked diff.

Reviewed By: saitcakmak

Differential Revision: D97148274

fbshipit-source-id: 146666b5ab93059ba39b2040e0a6a69cd972703b

Author: David Eriksson

botorch/models/fully_bayesian.py

@@ -809,6 +809,7 @@ class AbstractFullyBayesianSingleTaskGP(ExactGP, BatchedMultiOutputGPyTorchModel
 
     _is_fully_bayesian = True
     _is_ensemble = True
+    _supports_batched_models = False
     _pyro_model_class: type[PyroModel] = PyroModel
 
     def __init__(

botorch/models/map_saas.py

@@ -455,6 +455,7 @@ def __init__(
 
 class EnsembleMapSaasSingleTaskGP(SingleTaskGP):
     _is_ensemble = True
+    _supports_batched_models = False
 
     def __init__(
         self,

botorch/models/multitask.py

@@ -146,6 +146,8 @@ class MultiTaskGP(ExactGP, MultiTaskGPyTorchModel, FantasizeMixin):
     different noise levels for the different tasks.
     """
 
+    _supports_batched_models = False
+
     def __init__(
         self,
         train_X: Tensor,

test/models/test_fully_bayesian.py

@@ -347,6 +347,9 @@ def _get_mcmc_samples(
                 mcmc_samples[k] = torch.rand(num_samples, 1, dim, **tkwargs)
         return mcmc_samples
 
+    def test_supports_batched_models(self) -> None:
+        self.assertFalse(self.model_cls._supports_batched_models)
+
     def test_raises(self) -> None:
         tkwargs = {"device": self.device, "dtype": torch.double}
         with self.assertRaisesRegex(

test/models/test_map_saas.py

@@ -541,6 +541,9 @@ def test_emsemble_map_saas(self) -> None:
                 self.assertIsInstance(model.outcome_transform, Standardize)
                 self.assertFalse(hasattr(model, "input_transform"))
 
+    def test_ensemble_map_saas_supports_batched_models(self) -> None:
+        self.assertFalse(EnsembleMapSaasSingleTaskGP._supports_batched_models)
+
     def test_ensemble_map_saas_validation(self) -> None:
         with self.assertRaisesRegex(ValueError, "Expected taus to be of shape"):
             EnsembleMapSaasSingleTaskGP(

test/models/test_multitask.py

@@ -126,6 +126,9 @@ def _gen_kronecker_model_and_data(model_kwargs=None, batch_shape=None, **tkwargs
 
 
 class TestMultiTaskGP(BotorchTestCase):
+    def test_supports_batched_models(self) -> None:
+        self.assertFalse(MultiTaskGP._supports_batched_models)
+
     def test_MultiTaskGP(self) -> None:
         bounds = torch.tensor([[0.0, 0.0], [1.0, 1.0]])
         for dtype, use_intf, use_octf, task_values, fixed_noise in zip(