Skip to content

Conversation

crusaderky
Copy link
Contributor

Late addition to #2089

@crusaderky
Copy link
Contributor Author

@shoyer sorry, late addition.
Note that currently split_every is ignored by dask (dask/dask#3471)

@crusaderky crusaderky changed the title xarray.dot to pass optimize, split_every, etc. to einsum xarray.dot to pass **kwargs to einsum May 4, 2018
with pytest.raises(TypeError):
xr.dot(da_a.to_dataset(name='da'), dims='a')
with pytest.raises(TypeError):
xr.dot(dims='a')

# einsum parameters
xr.dot(da_a, da_b, dims=['b'], order='F')
actual = xr.dot(da_a, da_b, dims=['b'])
Copy link
Member

Choose a reason for hiding this comment

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

I'm a little confused here -- the result of the first line is thrown away rather than checked, and the second line doesn't use parameters at all?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

@shoyer Sorry, you're right it did not make much sense. Fixed.

assert actual.values.flags['F_CONTIGUOUS']


# einsum has a constant string as of the first parameter, which makes
Copy link
Contributor

Choose a reason for hiding this comment

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

E303 too many blank lines (2)

@crusaderky
Copy link
Contributor Author

@shoyer ready for merge if there are no further objections

@shoyer shoyer merged commit 218ad54 into pydata:master May 14, 2018
@shoyer
Copy link
Member

shoyer commented May 14, 2018

thanks!

@crusaderky crusaderky deleted the dask_einsum branch May 17, 2018 13:54
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.

4 participants