You can pass the dtype to zeros right here so you don't have to cast to int later, ie `labels = np.zeros((3, 3, 3), dtype=int)`.