Support sampler argument in LightningDataModule

[rusty1s]

No description provided.

[codecov]

Codecov Report

Merging #5456 (565f8ba) into master (67b8324) will decrease coverage by 2.08%.
The diff coverage is 5.26%.

@@            Coverage Diff             @@
##           master    #5456      +/-   ##
==========================================
- Coverage   85.42%   83.34%   -2.09%     
==========================================
  Files         348      348              
  Lines       18913    18935      +22     
==========================================
- Hits        16157    15781     -376     
- Misses       2756     3154     +398     

Impacted Files	Coverage Δ
torch_geometric/data/lightning_datamodule.py	43.14% <5.26%> (-5.98%)	⬇️
torch_geometric/nn/models/dimenet_utils.py	0.00% <0.00%> (-75.52%)	⬇️
torch_geometric/nn/models/dimenet.py	14.51% <0.00%> (-53.00%)	⬇️
torch_geometric/profile/profile.py	37.89% <0.00%> (-26.32%)	⬇️
torch_geometric/nn/conv/utils/typing.py	81.25% <0.00%> (-17.50%)	⬇️
torch_geometric/nn/inits.py	67.85% <0.00%> (-7.15%)	⬇️
torch_geometric/nn/pool/asap.py	92.10% <0.00%> (-6.58%)	⬇️
torch_geometric/transforms/add_self_loops.py	94.44% <0.00%> (-5.56%)	⬇️
torch_geometric/nn/resolver.py	88.88% <0.00%> (-5.56%)	⬇️
torch_geometric/sampler/neighbor_sampler.py	92.90% <0.00%> (-5.17%)	⬇️
... and 11 more

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