reformated with pre-commit hooks.

Author: jvreca

src/qonnx/custom_op/general/quant.py

@@ -142,18 +142,24 @@ def resolve_rounding_mode(mode_string):
     elif normalized_mode_string == "FLOOR":
         return np.floor
     elif normalized_mode_string == "UP":
+
         def round_up(x):
             return np.sign(x) * np.ceil(np.abs(x))
+
         return round_up
     elif normalized_mode_string == "DOWN":
         return np.fix
     elif normalized_mode_string == "HALF_UP":
+
         def round_half_up(x):
             return np.sign(x) * np.floor(np.abs(x) + 0.5)
+
         return round_half_up
     elif normalized_mode_string == "HALF_DOWN":
+
         def round_half_down(x):
             return np.sign(x) * np.ceil(np.abs(x) - 0.5)
+
         return round_half_down
     else:
         raise ValueError(f"Could not resolve rounding mode called: {normalized_mode_string}")

tests/custom_op/test_runding_mode.py

@@ -4,15 +4,18 @@
 
 from qonnx.custom_op.general.quant import resolve_rounding_mode
 
-@pytest.mark.parametrize("rmode,exp", [
+
+@pytest.mark.parametrize(
+    "rmode,exp",
+    [
         ("ROUND", np.array([6, 2, 2, 1, 1, -1, -1, -2, -2, -6])),
-        ("CEIL", np.array([6, 3, 2, 2, 1, -1, -1, -1, -2, - 5])),
+        ("CEIL", np.array([6, 3, 2, 2, 1, -1, -1, -1, -2, -5])),
         ("FLOOR", np.array([5, 2, 1, 1, 1, -1, -2, -2, -3, -6])),
         ("UP", np.array([6, 3, 2, 2, 1, -1, -2, -2, -3, -6])),
         ("DOWN", np.array([5, 2, 1, 1, 1, -1, -1, -1, -2, -5])),
         ("HALF_UP", np.array([6, 3, 2, 1, 1, -1, -1, -2, -3, -6])),
-        ("HALF_DOWN", np.array([5, 2, 2, 1, 1, -1, -1, -2, -2, -5]))
-    ]
+        ("HALF_DOWN", np.array([5, 2, 2, 1, 1, -1, -1, -2, -2, -5])),
+    ],
 )
 def test_rounding_modes(rmode, exp):
     test_array = np.array([5.5, 2.5, 1.6, 1.1, 1.0, -1.0, -1.1, -1.6, -2.5, -5.5])