Refactor NeighborSampler to be input-type agnostic

[rusty1s]

No description provided.

[codecov]

Codecov Report

Merging #6173 (9603aa5) into master (15b7a4c) will increase coverage by 0.00%.
The diff coverage is 93.38%.

@@           Coverage Diff           @@
##           master    #6173   +/-   ##
=======================================
  Coverage   84.37%   84.37%           
=======================================
  Files         366      366           
  Lines       20559    20544   -15     
=======================================
- Hits        17346    17335   -11     
+ Misses       3213     3209    -4     

Impacted Files	Coverage Δ
torch_geometric/data/lightning/datamodule.py	55.76% <ø> (-0.15%)	⬇️
torch_geometric/loader/hgt_loader.py	100.00% <ø> (ø)
torch_geometric/loader/link_neighbor_loader.py	100.00% <ø> (ø)
torch_geometric/loader/link_loader.py	76.71% <61.53%> (ø)
torch_geometric/sampler/neighbor_sampler.py	66.53% <95.45%> (-0.27%)	⬇️
torch_geometric/loader/node_loader.py	87.35% <100.00%> (-0.29%)	⬇️
torch_geometric/loader/utils.py	83.33% <100.00%> (-0.86%)	⬇️
torch_geometric/sampler/__init__.py	100.00% <100.00%> (ø)
torch_geometric/sampler/base.py	95.77% <100.00%> (+2.02%)	⬆️
torch_geometric/sampler/hgt_sampler.py	100.00% <100.00%> (ø)
... and 1 more

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more