from torch_geometric_temporal.signal import DynamicGraphTemporalSignal
dataset = DynamicGraphTemporalSignal(
            edge_indices, edge_features, xs, ys, y_indices=y_indices
        )

i = 4
dataset[i]

Data(x=[603, 2], edge_index=[2, 1946], edge_attr=[1946, 3], y=[13], y_indices=[2, 13])