Linting for dilations pt 1

cog-imperial · Nov 17, 2023 · 4a83a22 · 4a83a22
1 parent 668f192
commit 4a83a22
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/src/omlt/neuralnet/layer.py b/src/omlt/neuralnet/layer.py
@@ -508,17 +508,17 @@ def __init__(
                                             kernel.shape[0],
                                             kernel.shape[1],
                                             1,
-                                            kernel.shape[3]
+                                            kernel.shape[3],
                                         )
                                     ),
                                     np.zeros(
                                         (
                                             kernel.shape[0],
                                             kernel.shape[1],
                                             self.dilations[0] - 1,
-                                            kernel.shape[3]
+                                            kernel.shape[3],
                                         )
-                                    )
+                                    ),
                                 ]
                             )
                             for i in range(kernel.shape[2] - 1)
@@ -540,24 +540,29 @@ def __init__(
                                             dilate_rows.shape[0],
                                             dilate_rows.shape[1],
                                             dilate_rows.shape[2],
-                                            1
+                                            1,
                                         )
                                     ),
                                     np.zeros(
                                         (
                                             dilate_rows.shape[0],
                                             dilate_rows.shape[1],
                                             dilate_rows.shape[2],
-                                            self.dilations[1] - 1
+                                            self.dilations[1] - 1,
                                         )
-                                    )
+                                    ),
                                 ]
                             )
-                            for i in range(dilate_rows.shape[3]-1)
+                            for i in range(dilate_rows.shape[3] - 1)
                         ]
                     ),
                     dilate_rows[:, :, :, -1].reshape(
-                        (dilate_rows.shape[0], dilate_rows.shape[1], dilate_rows.shape[2], 1)
+                        (
+                            dilate_rows.shape[0],
+                            dilate_rows.shape[1],
+                            dilate_rows.shape[2],
+                            1,
+                        )
                     ),
                 ]
             )