Refactor NeighborSampler
to be input-type agnostic
#6173
+289
−307
Merged