import torch as to

BASE_GRAPH = to.tensor([[0, 1, 1, 0],
                        [1, 0, 1, 0],
                        [1, 1, 0, 1],
                        [0, 0, 1, 0]])
BASE_GRAPH_NODE_FEATURES = to.tensor([[1, 2], [1, 1], [2, 0.5], [0.5, 0.5]])
BASE_GRAPH_EDGE_FEATURES = to.tensor([[[0.0, 0.0], [1.0, 2.0], [2.0, 0.5], [0.0, 0.0]],
                                      [[1.0, 2.0], [0.0, 0.0], [1.0, 1.0], [0.0, 0.0]],
                                      [[2.0, 0.5], [1.0, 1.0], [0.0, 0.0], [0.5, 0.5]],
                                      [[0.0, 0.0], [0.0, 0.0], [0.5, 0.5], [0.0, 0.0]]])

torch.tensor