TypeError: forward() takes 6 positional arguments but 9 were given(maybe something wrong with the source codes?

[Milkshade]

here seems a mistake in the /softgroup/ops/functions.py , in line 265 have a line
nActive = ops.ballquery_batch_p(coords, batch_idxs, batch_offsets, idx, start_len, n,
meanActive, radius)
but the forward function class BallQueryBatchP(Function):
@staticmethod
def forward(ctx, coords, batch_idxs, batch_offsets, radius, meanActive):
only have 6 inputs, when i run the codes , there appears a mistake:TypeError: forward() takes 6 positional arguments but 9 were given

[thangvubk]

ops.ballquery_batch_p is the cuda function and the argument is listed in here

SoftGroup/softgroup/ops/src/bfs_cluster/bfs_cluster.h

Line 14 in 11dcbfd

int ballquery_batch_p(at::Tensor xyz_tensor, at::Tensor batch_idxs_tensor,