Ensure input is 'dstack' format