db830daee51fa17726524533da469fb91804b2a0,gpytorch/lazy/interpolated_lazy_variable.py,InterpolatedLazyVariable,root_decomposition,#InterpolatedLazyVariable#,349

Before Change

                              self.right_interp_values.repeat(*sizes), **self._kwargs)

    def root_decomposition(self):
        if isinstance(self.base_lazy_variable, RootLazyVariable):
            interp_root = InterpolatedLazyVariable(self.base_lazy_variable.root, self.left_interp_indices,
                                                   self.left_interp_values)
            return RootLazyVariable(interp_root)
        else:
            super(InterpolatedLazyVariable, self).root_decomposition()

    def root_decomposition_size(self):
        if isinstance(self.base_lazy_variable, RootLazyVariable):
            return self.base_lazy_variable.root_decomposition_size()
        else:

After Change

                              self.right_interp_values.repeat(*sizes), **self._kwargs)

    def root_decomposition(self):
        base_root_decomposition = self.base_lazy_variable.root_decomposition().root
        root_interp_lazy_variable = self.__class__(base_root_decomposition, self.left_interp_indices,
                                                   self.left_interp_values)
        return RootLazyVariable(root_interp_lazy_variable)

Instances