update neighbor sample API #5913

Job	Run time
devcontainer / build (amd64, 12.5, conda)	1h 17m 0s
wheel-build-cugraph-equivariant / compute-matrix	2s
checks / other-checks	35s
checks / check-style	1m 57s
devcontainer / build (amd64, 12.5, pip)	1h 21m 13s
wheel-build-cugraph-equivariant / 11.8.0, 3.10, amd64, rockylinux8	3m 13s
wheel-build-cugraph-equivariant / 12.5.1, 3.10, amd64, rockylinux8	3m 26s
wheel-build-cugraph-equivariant / 11.8.0, 3.11, amd64, rockylinux8	3m 10s
wheel-build-cugraph-equivariant / 12.5.1, 3.11, amd64, rockylinux8	3m 14s
wheel-build-cugraph-equivariant / 11.8.0, 3.10, arm64, rockylinux8	2m 42s
wheel-build-cugraph-equivariant / 12.5.1, 3.10, arm64, rockylinux8	2m 46s
wheel-build-cugraph-equivariant / 11.8.0, 3.11, arm64, rockylinux8	2m 34s
wheel-build-cugraph-equivariant / 12.5.1, 3.11, arm64, rockylinux8	2m 46s
conda-cpp-build	0s
wheel-build-pylibcugraph	0s
conda-cpp-checks	0s
conda-python-build	0s
conda-cpp-tests	0s
wheel-tests-pylibcugraph	0s
conda-notebook-tests	0s
conda-python-tests	0s
docs-build	0s
wheel-build-cugraph	0s
wheel-build-nx-cugraph	0s
wheel-tests-cugraph	0s
wheel-tests-nx-cugraph	0s
wheel-build-cugraph-dgl	0s
wheel-build-cugraph-pyg	0s
wheel-tests-cugraph-dgl	0s
wheel-tests-cugraph-pyg	0s
wheel-tests-cugraph-equivariant / compute-matrix	9s
wheel-tests-cugraph-equivariant / 12.5.1, 3.10, amd64, ubuntu22.04, v100, latest-driver, oldest-deps	6m 59s
pr-builder	0s
	3h 11m 46s

Provide feedback