fix format

Author: PierreMarchand20

example/define_custom_local_operator.py

@@ -9,7 +9,7 @@ def __init__(
         target_cluster: Htool.Cluster,
         global_source_cluster: Htool.Cluster,
         local_source_offset: int,
-        local_source_size:int,
+        local_source_size: int,
     ) -> None:
         super().__init__(
             target_cluster,
@@ -22,12 +22,10 @@ def __init__(
         self.data = np.zeros((target_size, local_source_size))
         generator.build_submatrix(
             target_cluster.get_permutation()[
-                target_offset : target_offset
-                + target_size
+                target_offset : target_offset + target_size
             ],
             global_source_cluster.get_permutation()[
-                local_source_offset : local_source_offset
-                + local_source_size
+                local_source_offset : local_source_offset + local_source_size
             ],
             self.data,
         )
@@ -38,20 +36,60 @@ def add_vector_product(
         # Beware, inplace operation needed for output to keep the underlying data
         output *= beta
         if trans == "N":
-            output += alpha * self.data.dot(input[self.local_source_offset : self.local_source_offset+ self.local_source_size])
+            output += alpha * self.data.dot(
+                input[
+                    self.local_source_offset : self.local_source_offset
+                    + self.local_source_size
+                ]
+            )
         elif trans == "T":
-            output += alpha * np.transpose(self.data).dot(input[self.local_source_offset : self.local_source_offset+ self.local_source_size])
+            output += alpha * np.transpose(self.data).dot(
+                input[
+                    self.local_source_offset : self.local_source_offset
+                    + self.local_source_size
+                ]
+            )
         elif trans == "C":
-            output += alpha * np.vdot(np.transpose(self.data), input[self.local_source_offset : self.local_source_offset+ self.local_source_size])
+            output += alpha * np.vdot(
+                np.transpose(self.data),
+                input[
+                    self.local_source_offset : self.local_source_offset
+                    + self.local_source_size
+                ],
+            )
 
     def add_matrix_product_row_major(
         self, trans, alpha, input: np.array, beta, output: np.array
     ) -> None:
         output *= beta
         if trans == "N":
-            output += alpha * self.data @ input[self.local_source_offset : self.local_source_offset+ self.local_source_size,:]
+            output += (
+                alpha
+                * self.data
+                @ input[
+                    self.local_source_offset : self.local_source_offset
+                    + self.local_source_size,
+                    :,
+                ]
+            )
         elif trans == "T":
-            output += alpha * np.transpose(self.data) @ input[self.local_source_offset : self.local_source_offset+ self.local_source_size,:]
+            output += (
+                alpha
+                * np.transpose(self.data)
+                @ input[
+                    self.local_source_offset : self.local_source_offset
+                    + self.local_source_size,
+                    :,
+                ]
+            )
         elif trans == "C":
-            output += alpha * np.matrix.H(self.data) @ input[self.local_source_offset : self.local_source_offset+ self.local_source_size,:]
+            output += (
+                alpha
+                * np.matrix.H(self.data)
+                @ input[
+                    self.local_source_offset : self.local_source_offset
+                    + self.local_source_size,
+                    :,
+                ]
+            )
         output = np.asfortranarray(output)

example/use_local_hmatrix_compression.py

@@ -83,13 +83,20 @@
     )
 
 local_operator_2 = None
-if source_cluster.get_size()-local_source_cluster.get_size()-local_source_cluster.get_offset() > 0:
+if (
+    source_cluster.get_size()
+    - local_source_cluster.get_size()
+    - local_source_cluster.get_offset()
+    > 0
+):
     local_operator_2 = CustomLocalOperator(
         generator,
         local_target_cluster,
         source_cluster,
-        local_source_cluster.get_size()+local_source_cluster.get_offset(),
-        source_cluster.get_size()-local_source_cluster.get_size()-local_source_cluster.get_offset(),
+        local_source_cluster.get_size() + local_source_cluster.get_offset(),
+        source_cluster.get_size()
+        - local_source_cluster.get_size()
+        - local_source_cluster.get_offset(),
     )
 
 if local_operator_1: