Fix

Signed-off-by: Serge Panev <[email protected]>

torch_geometric/sampler/neighbor_sampler.py

@@ -159,7 +159,7 @@ def __init__(
             def hook_func_with_nvtx(func_name):
                 func = getattr(self, func_name)
 
-                def hooked_func(**kwargs):
+                def hooked_func(*args, **kwargs):
                     nvtx_handle = torch.cuda.nvtx.range_start(
                         f"[Sampler] {func_name} for {self}")
                     ret = func(**kwargs)