Explicitly specify int64 since default dtype is different on e.g. Windows