5272720e291d084e775df43a4e1920fedb48887b,gpytorch/models/exact_gp.py,ExactGP,get_fantasy_model,#ExactGP#Any#Any#,85

Before Change

                "all test independent caches exist. Call the model on some data first!"
            )

        if self.train_inputs[0].dim() > 2:
            raise RuntimeError(
                "Adding fantasy observations to a GP that is already batch mode is currently unsupported."
            )

        if self.train_targets.dim() > 1:
            raise RuntimeError("Cannot yet add fantasy observations to multitask GPs, but this is coming soon!")

        if not isinstance(inputs, list):

After Change

        train_targets = self.train_targets.expand(target_batch_shape + self.train_targets.shape[-1:])

        full_inputs = [
            torch.cat([train_input, input.expand(input_batch_shape + input.shape[-2:])], dim=-2)
            for train_input, input in zip(train_inputs, inputs)
        ]
        full_targets = torch.cat([train_targets, targets.expand(target_batch_shape + targets.shape[-1:])], dim=-1)

Instances