From da6c316eea84e9679dec0d8784b60da0a5c32e53 Mon Sep 17 00:00:00 2001
From: Jeremy Sadler <53983960+jezsadler@users.noreply.github.com>
Date: Mon, 22 Apr 2024 00:35:06 +0000
Subject: [PATCH] Fixing some whitespace linting

---
 src/omlt/neuralnet/layer.py            |  3 ++-
 tests/neuralnet/test_nn_formulation.py | 15 +++++++++++----
 2 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/src/omlt/neuralnet/layer.py b/src/omlt/neuralnet/layer.py
index fd1b3234..16e068a3 100644
--- a/src/omlt/neuralnet/layer.py
+++ b/src/omlt/neuralnet/layer.py
@@ -16,6 +16,7 @@
     \end{align*}
 
 """
+
 import itertools
 
 import numpy as np
@@ -254,7 +255,7 @@ class GNNLayer(DenseLayer):
     .. math::
 
         \begin{align*}
-            y_j = \sigma \left(\sum\limits_{i=0}^{F_{in}-1}A_{u,v}w_{ij}x_i+b_j\right), && \forall 0\le j<F_{out}, 
+            y_j = \sigma \left(\sum\limits_{i=0}^{F_{in}-1}A_{u,v}w_{ij}x_i+b_j\right), && \forall 0\le j<F_{out},
         \end{align*}
 
 
diff --git a/tests/neuralnet/test_nn_formulation.py b/tests/neuralnet/test_nn_formulation.py
index 2e1d6cb5..e8d54068 100644
--- a/tests/neuralnet/test_nn_formulation.py
+++ b/tests/neuralnet/test_nn_formulation.py
@@ -544,12 +544,16 @@ def test_partition_based_unbounded_below():
     m.neural_net_block.build_formulation(formulation)
     prev_layer_block = m.neural_net_block.layer[prev_layer_id]
     prev_layer_block.z.setlb(-interval.inf)
-    
+
     split_func = lambda w: default_partition_split_func(w, 2)
 
     with pytest.raises(ValueError) as excinfo:
         partition_based_dense_relu_layer(
-            m.neural_net_block, net, m.neural_net_block.layer[test_layer_id], test_layer, split_func
+            m.neural_net_block,
+            net,
+            m.neural_net_block.layer[test_layer_id],
+            test_layer,
+            split_func,
         )
     expected_msg = "Expression is unbounded below."
     assert str(excinfo.value) == expected_msg
@@ -568,12 +572,15 @@ def test_partition_based_unbounded_above():
     prev_layer_block = m.neural_net_block.layer[prev_layer_id]
     prev_layer_block.z.setub(interval.inf)
 
-
     split_func = lambda w: default_partition_split_func(w, 2)
 
     with pytest.raises(ValueError) as excinfo:
         partition_based_dense_relu_layer(
-            m.neural_net_block, net, m.neural_net_block.layer[test_layer_id], test_layer, split_func
+            m.neural_net_block,
+            net,
+            m.neural_net_block.layer[test_layer_id],
+            test_layer,
+            split_func,
         )
     expected_msg = "Expression is unbounded above."
     assert str(excinfo.value) == expected_msg