-
-
Notifications
You must be signed in to change notification settings - Fork 1.2k
xarray.dot to pass **kwargs to einsum #2106
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
Conversation
@shoyer sorry, late addition. |
xarray/tests/test_computation.py
Outdated
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']) |
There was a problem hiding this comment.
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?
There was a problem hiding this comment.
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.
xarray/tests/test_computation.py
Outdated
assert actual.values.flags['F_CONTIGUOUS'] | ||
|
||
|
||
# einsum has a constant string as of the first parameter, which makes |
There was a problem hiding this comment.
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)
@shoyer ready for merge if there are no further objections |
thanks! |
Late addition to #2089