631d4ee2ece28da70d6e22201129f39b4f1dc586,torch_geometric/data/temporal.py,TemporalData,__getitem__,#TemporalData#Any#,17

Before Change

            mask = self.t == idx
        elif isinstance(idx, slice):
            start = idx.start if idx.start is not None else int(self.t.min())
            stop = idx.stop if idx.stop is not None else int(self.t.max() + 1)
            assert idx.step is None and start <= stop
            mask = (self.t >= start) & (self.t < stop)
        else:
            raise IndexError(
                f"Only integers, slices (`:`), and strings are valid indices "
                f"(got {type(idx).__name__}).")

After Change

        if isinstance(idx, int):
            idx = torch.tensor([idx])
        if isinstance(idx, (list, tuple)):
            idx = torch.tensor(idx)
        elif isinstance(idx, slice):
            pass
        elif isinstance(idx, torch.Tensor) and (idx.dtype == torch.long
                                                or idx.dtype == torch.bool):

Instances