Question about Data Import and relation representation

[benearnthof]

Hi, first of all thanks for providing this code, it is very nice to work with and I was able to reproduce 88% accuracy within 15 minutes of tinkering with the repository, very impressive!

I went to the website for the aifb dataset and was wondering how you converted the graph you get from importing the data to the array of sparse matrices that is contained in the pickle file you provide in this repo. Would really help me understand how to encode my own data to experiment with this architecture on different datasets. Can you provide code for dataset conversion to sparse matrices or recommend any packages you used? Best regards!