Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Sep 22, 2023
1 parent edac832 commit 3449994
Show file tree
Hide file tree
Showing 5 changed files with 8 additions and 4 deletions.
5 changes: 4 additions & 1 deletion src/amrex/MultiFab.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
License: BSD-3-Clause-LBNL
"""


def mf_to_numpy(amr, self, copy=False, order="F"):
"""
Provide a Numpy view into a MultiFab.
Expand Down Expand Up @@ -92,7 +93,9 @@ def register_MultiFab_extension(amr):
"""MultiFab helper methods"""

# register member functions for the MultiFab type
amr.MultiFab.to_numpy = lambda self, copy=False, order="F": mf_to_numpy(amr, self, copy, order)
amr.MultiFab.to_numpy = lambda self, copy=False, order="F": mf_to_numpy(
amr, self, copy, order
)
amr.MultiFab.to_numpy.__doc__ = mf_to_numpy.__doc__

amr.MultiFab.to_cupy = mf_to_cupy
2 changes: 1 addition & 1 deletion src/amrex/space1d/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,8 +45,8 @@ def Print(*args, **kwargs):


from ..Array4 import register_Array4_extension
from ..MultiFab import register_MultiFab_extension
from ..ArrayOfStructs import register_AoS_extension
from ..MultiFab import register_MultiFab_extension
from ..PODVector import register_PODVector_extension
from ..StructOfArrays import register_SoA_extension

Expand Down
2 changes: 1 addition & 1 deletion src/amrex/space2d/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,8 +45,8 @@ def Print(*args, **kwargs):


from ..Array4 import register_Array4_extension
from ..MultiFab import register_MultiFab_extension
from ..ArrayOfStructs import register_AoS_extension
from ..MultiFab import register_MultiFab_extension
from ..PODVector import register_PODVector_extension
from ..StructOfArrays import register_SoA_extension

Expand Down
2 changes: 1 addition & 1 deletion src/amrex/space3d/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,8 +45,8 @@ def Print(*args, **kwargs):


from ..Array4 import register_Array4_extension
from ..MultiFab import register_MultiFab_extension
from ..ArrayOfStructs import register_AoS_extension
from ..MultiFab import register_MultiFab_extension
from ..PODVector import register_PODVector_extension
from ..StructOfArrays import register_SoA_extension

Expand Down
1 change: 1 addition & 0 deletions tests/test_multifab.py
Original file line number Diff line number Diff line change
Expand Up @@ -357,5 +357,6 @@ def test_mfab_dtoh_copy(make_mfab_device):

# cupy bindings (w/o copy)
import cupy as cp

local_boxes_device = mfab_device.to_cupy()
assert max([cp.max(box) for box in local_boxes_device]) == device_max

0 comments on commit 3449994

Please sign in to comment.