Add warning message when no output_shape or output_type is missing