Error when running compute_transition_matrix()

I was trying to run compute_transition_matrix() on a subset with less than 1000 cells (960) and keep getting the error: “NetworkxError: Adjacency matrix not square”
This happens only if the adata has les than 1000 cells. Would anyone know how to get around this?
Thanks in advance