Refactored script to save results as fits files.

import os

from radmc3dPy import image

from typing import Union

from assets.commons import validate_parameter

def main(cube_out_name: Union[str, None] = None,

         cube_out_path: Union[str, None] = None):

    _cube_out_name = validate_parameter(cube_out_name, default='test_cube.fits')

    _cube_out_path = validate_parameter(cube_out_path, default=os.path.join('prs', 'fits', 'cubes'))

    imdata = image.readImage(fname='mdl/radmc_files/image.out')

imdata.writeFits(fname='prs/fits/cubes/test_cube.fits')

    imdata.writeFits(fname=os.path.join(_cube_out_path, _cube_out_name))

if __name__ == '__main__':

    main()