IndexError: index ... is out of bounds for axis 0 with size ...

[alirezabayatmk]

I'm getting the same error.

Traceback (most recent call last):
  File "/home/shady/Projects/GML/SPLITTER/Splitter/src/main.py", line 24, in <module>
    main()
  File "/home/shady/Projects/GML/SPLITTER/Splitter/src/main.py", line 19, in main
    trainer.fit()
  File "/home/shady/Projects/GML/SPLITTER/Splitter/src/splitter.py", line 229, in fit
    self.setup_model()
  File "/home/shady/Projects/GML/SPLITTER/Splitter/src/splitter.py", line 159, in setup_model
    self.egonet_splitter.personality_map)
  File "/home/shady/Projects/GML/SPLITTER/Splitter/src/splitter.py", line 52, in initialize_weights
    persona_embedding = np.array([base_node_embedding[n] for _, n in mapping.items()])
  File "/home/shady/Projects/GML/SPLITTER/Splitter/src/splitter.py", line 52, in <listcomp>
    persona_embedding = np.array([base_node_embedding[n] for _, n in mapping.items()])
IndexError: index 8637 is out of bounds for axis 0 with size 8637

The dataset is sorted and also IDs start from zero with no index and header. Also, it just happens on the CA-HepTh dataset and not others which is strange.