minor: fix linting

PyLops · Oct 23, 2024 · 9786dbd · 9786dbd
1 parent 7a5ea19
commit 9786dbd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pylops_mpi/basicoperators/BlockDiag.py b/pylops_mpi/basicoperators/BlockDiag.py
@@ -1,7 +1,7 @@
 import numpy as np
 from scipy.sparse.linalg._interface import _get_dtype
 from mpi4py import MPI
-from typing import Optional, Sequence, Union, List
+from typing import Optional, Sequence, List
 from numbers import Integral
 
 from pylops import LinearOperator

diff --git a/tests/test_distributedarray.py b/tests/test_distributedarray.py
@@ -199,7 +199,7 @@ def test_distributed_norm(par):
                                         (par8, par9), (par8b, par9b)])
 def test_distributed_maskeddot(par1, par2):
     """Test Distributed Dot product with masked array"""
-    nsub = 3 # number of subcommunicators
+    nsub = 3  # number of subcommunicators
     subsize = max(1, MPI.COMM_WORLD.Get_size() // nsub)
     mask = np.repeat(np.arange(nsub), subsize)
     # Replicate x1 and x2 as required in masked arrays