I could just do np.sqrt on the count in the sem method, but I felt this approach was more robust to future changes to the private API, and would allow for potential optimizations down the road if anyone wants to do them.