black>=22.0
numpy
torch
torch_geometric
