issue_comments

also pinging @dcherian who has been working on a similar problem set with map_blocks in #3816.

still relevant

dtype = [getattr(x, 'dtype', getattr(x, 'dtypes'))] would be another alternative, but I agree it's ugly. The ternary expression dtype = x.dtypes if isinstance(x, xarray.Dataset) else x.dtype would also work.

I agree with the concern about duck typing, but my concern with Dataset.dtype is that there is strong convention for a dtype attribute to be an actual NumPy dtype.

Another option would be accept either objects with a dtype or dtypes in output_dtypes, like np.result_type(). Then you could write output_dtype=[x].

The key thing is that for most people it would be extremely elegant and practical to be able to duck-type wrappers around numpy, scipy, and numba kernels that automagically work with Variable, DataArray, and Dataset (see my example above). You'll agree on how ugly my 1-liner above would become: def myfunc(x): if isinstance(x, xarray.Dataset): dtype = x.dtypes else: # DataArray and Variable dtype = x.dtype return apply_ufunc(numpy_kernel, x, dask='parallelized', output_dtypes=[dtype]) If you don't like Dataset.dtype, then maybe we could add both Dataset.dtypes and DataArray.dtypes (which would be just an alias to DataArray.dtype)? I still like the former more though - I find it less confusing.

I'm not sure about adding Dataset.dtype. Certainly Dataset.dtypes returning a dict would make sense -- that would match how pandas defines DataFrame.dtypes.

Anyways, I agree that output_dtypes=[{var1: t1, var2: t2, ...}, ...] is most natural, because it matches the structure of the outputs.

@shoyer that seems counter-intuitive for me - you are returning two datasets after all. If we go with the list(dict) notation, we could also add a Dataset.dtype property, which (coherently with dims and chunks) would return a dict. This would be very handy as, in 99% of the times, people will want to write: def myfunc(x): return apply_ufunc(numpy_kernel, x, dask='parallelized', output_dtypes=[x.dtype]) which would magically work both when x is a DataArray and when it's a Dataset

Yes, I like this. Though it's worth considering whether the syntax should reverse the list/dict nesting, e.g., output_dtypes={var1: [t1, ...], var2: [t2, ...], ...}.