add a `TODO: add stokes axis on writing for CASA compatibility`