Skip to content

Commit

Permalink
Fix a name in TransferFromNumpyMapper
Browse files Browse the repository at this point in the history
  • Loading branch information
inducer committed Nov 29, 2024
1 parent 404d4a1 commit 64ef2f5
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions arraycontext/impl/pytato/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -158,13 +158,13 @@ def map_data_wrapper(self, expr: DataWrapper) -> Array:
# Ideally, this code should just do
# return self.actx.from_numpy(expr.data).tagged(expr.tags),
# but there seems to be no way to transfer the non_equality_tags in that case.
new_dw = self.actx.from_numpy(expr.data)
assert isinstance(new_dw, DataWrapper)
actx_ary = self.actx.from_numpy(expr.data)
assert isinstance(actx_ary, DataWrapper)

# https://github.com/pylint-dev/pylint/issues/3893
# pylint: disable=unexpected-keyword-arg
return DataWrapper(
data=new_dw.data,
data=actx_ary.data,
shape=expr.shape,
axes=expr.axes,
tags=expr.tags,
Expand Down

0 comments on commit 64ef2f5

Please sign in to comment.