Skip to content

Commit 02efdbb

Browse files
committed
minor: fix more flake8
1 parent 83f7a8b commit 02efdbb

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pylops_mpi/utils/_mpi.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,7 @@ def mpi_send(base_comm: MPI.Comm,
167167
Tag of the message to be sent.
168168
engine : :obj:`str`, optional
169169
Engine used to store array (``numpy`` or ``cupy``)
170-
170+
171171
"""
172172
if deps.cuda_aware_mpi_enabled or engine == "numpy":
173173
# Determine MPI type based on array dtype
@@ -183,7 +183,7 @@ def mpi_send(base_comm: MPI.Comm,
183183
def mpi_recv(base_comm: MPI.Comm,
184184
recv_buf: Optional[NDArray] = None,
185185
source: int = 0,
186-
count: Optional[int] = None,
186+
count: Optional[int] = None,
187187
tag: int = 0,
188188
engine: Optional[str] = "numpy",
189189
) -> NDArray:
@@ -205,7 +205,7 @@ def mpi_recv(base_comm: MPI.Comm,
205205
Tag of the message to be sent.
206206
engine : :obj:`str`, optional
207207
Engine used to store array (``numpy`` or ``cupy``)
208-
208+
209209
Returns
210210
-------
211211
recv_buf : :obj:`numpy.ndarray` or :obj:`cupy.ndarray`

0 commit comments

Comments
 (0)