snap-stanford / GraphGym

Platform for designing and evaluating Graph Neural Networks (GNN)
Other
1.69k stars 185 forks source link

use float instead of int for fill_value #33

Closed SCAuFish closed 2 years ago

SCAuFish commented 2 years ago

In torch-geometry version 2.0.2, which is the version automatically installed with pip 21.1.3, add_remaining_self_loops accepts fill_value as one of float, tensor or str. Although in latest torch-geometric package, int value is accepted and treated the same way as float, it could be more robust to set fill_value as float.