GraphSAGE requires neighbor sampling, which is provided by either torch-sparse or (optimally) pyg-lib. It also throws a warning for missing torch-scatter but I think that might be optional for the functionality we need.
However, these libraries rely on CUDA and have historically raised conflicts with other versions of things we need when I tried to install them on Snellius. I have raised the issue with SURF - if they can find a good solution, we can either attempt to recreate it on the OSSC ourselves or export the virtual environment again.
GraphSAGE requires neighbor sampling, which is provided by either torch-sparse or (optimally) pyg-lib. It also throws a warning for missing torch-scatter but I think that might be optional for the functionality we need.
However, these libraries rely on CUDA and have historically raised conflicts with other versions of things we need when I tried to install them on Snellius. I have raised the issue with SURF - if they can find a good solution, we can either attempt to recreate it on the OSSC ourselves or export the virtual environment again.