Skip to content

Support dot with older dask #2205

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Jun 1, 2018

Conversation

fujiisoup
Copy link
Member

Related with #2203, I think it is better if xr.DataArray.dot() is working even with older dask, at least in the simpler case (as this is a very primary operation).

The cost is a slight complication of the code.
Any comments are welcome.

Copy link
Member

@shoyer shoyer left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This seems like a good idea to me 👍

@@ -784,6 +781,10 @@ def test_dot(use_dask):
assert (actual.data == np.einsum('ij,ijk->k', a, b)).all()
assert isinstance(actual.variable.data, type(da_a.variable.data))

import dask
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This needs to be guarded behind use_dask. Currently the tests without dask installed are failing here on Travis-CI.

@@ -37,6 +37,10 @@ Documentation
Enhancements
~~~~~~~~~~~~

- `:py:meth:`~DataArray.dot` and :py:func:`~dot` are partly supported with older
dask<01.7.4. (related to :issue:`2203`)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

01.7.4 -> 0.17.4

@fujiisoup fujiisoup merged commit 9d60897 into pydata:master Jun 1, 2018
@fujiisoup fujiisoup deleted the tensordot_with_olderdask branch June 1, 2018 01:01
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants