Since entities can be edges, facets or cells, it would return (num entities, num nodes per entity). Since your case is a vertex, there is only one node (1-1 map).
I would call:
idx = entities_to_geometry(mesh, 0, entity_idx).flatten()
Rather than doing it after