Indices not in valid range (HeteroData)

[nakhlarafi]

Hello there,

I was trying to make and run a heterogeneous graph in a model. However, I am getting errors like:

Encountered an index error. Please ensure that all indices in 'edge_index' point to valid indices in the interval [0, 571] (got interval [26, 974])

Now I manually checked the graphs and found that all indices are in the correct range. For example, these are some graph nodes:

Line Nodes:
(0, {'node': 8, 'proj': 'Lang29'})
(1, {'node': 8, 'proj': 'Lang29'})
(2, {'node': 7, 'proj': 'Lang29'})
(3, {'node': 8, 'proj': 'Lang29'})
(4, {'node': 7, 'proj': 'Lang29'})
(5, {'node': 6, 'proj': 'Lang29'})
(6, {'node': 6, 'proj': 'Lang29'})
(7, {'node': 8, 'proj': 'Lang29'})
(8, {'node': 8, 'proj': 'Lang29'})
(9, {'node': 6, 'proj': 'Lang29'})

Test Nodes:
(10, {'node': 3, 'proj': 'Lang29', 'textb': ['system', 'utils', 'test', 'test', 'java', 'version', 'as', 'int'], 'outcome': 1})
(11, {'node': 5, 'proj': 'Lang29', 'outcome': 0})

Method Nodes:
(12, {'node': 2, 'res': 0, 'proj': 'Lang29', 'overlap': 0.0, 'modification': 0.0})
(13, {'node': 2, 'res': 0, 'proj': 'Lang29', 'overlap': 0.2857142857142857, 'modification': 0.0})
(14, {'node': 2, 'res': 1, 'proj': 'Lang29', 'overlap': 1.0, 'modification': 1.0})

and these are the edges between them:

'method', 'has_line', 'line' tensor([[13, 12, 13, 13, 13, 12, 13, 13, 13, 14],
        [ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9]])
'line', 'outcome', 'test' tensor([[ 0,  1,  1,  2,  3,  4,  5,  5,  6,  7,  8,  9],
        [10, 11, 10, 10, 10, 10, 11, 10, 10, 10, 10, 10]])

Now can anyone tell me if the edge indices are in the correct range or if I am doing anything wrong?

This is the HeteroData Obj:

HeteroData(
  line={ x=[10, 1] },
  test={ x=[2, 2] },
  method={
    x=[3, 3],
    y=[3],
  },
  (method, has_line, line)={
    edge_index=[2, 10],
    edge_attr=[10],
  },
  (line, outcome, test)={
    edge_index=[2, 12],
    edge_attr=[12],
  }
)

[wsad1]

The node indices of need to go from 0-num_nodes-1 for each node type. In your example the indices of the Test Node type should be 0 and 1, and the indices of the Method Node should be 0, 1, 2, so your edge_index also needs to be updated accordingly.

[nakhlarafi]

Thanks! Now I get it.

One more question though. Does the DataLoader works on heterogeneous graphs?

[wsad1]

Yes it does.