Fix fit extra parameters

Allowed extra fit parameters to be non-array (e.g. str).

Added a tag to prevent stacking of the input array of the fit method if it expects partitioned data.