Pre-commit formatting

benchmarks/python/single_ops.py

@@ -50,12 +50,14 @@ def time_maximum():
     mx.eval(a, b)
     time_fn(mx.maximum, a, b)
 
+
 def time_max():
     a = mx.random.uniform(shape=(32, 1024, 1024))
     a[1, 1] = mx.nan
     mx.eval(a)
     time_fn(mx.max, a, 0)
 
+
 def time_negative():
     a = mx.random.uniform(shape=(10000, 1000))
     mx.eval(a)

mlx/backend/metal/kernels/reduction/ops.h

@@ -215,14 +215,12 @@ struct Max {
     }
   }
 
-
   template <>
   complex64_t operator()(complex64_t a, complex64_t b) {
-    if (metal::isnan(a.real) || metal::isnan(a.imag) || metal::isnan(b.real) || metal::isnan(b.imag)) {
+    if (metal::isnan(a.real) || metal::isnan(a.imag) || metal::isnan(b.real) ||
+        metal::isnan(b.imag)) {
       return static_cast<complex64_t>(NAN);
     }
     return a > b ? a : b;
-
   }
-
 };

python/tests/test_reduce.py

@@ -179,5 +179,6 @@ class TestReduce(mlx_tests.MLXTestCase):
                         ref = getattr(np, op)(x_np, axis=axis)
                         self.assertTrue(np.array_equal(out, ref, equal_nan=True))
 
+
 if __name__ == "__main__":
     mlx_tests.MLXTestRunner(failfast=True)

tests/ops_tests.cpp

@@ -1027,7 +1027,6 @@ TEST_CASE("test reduction ops") {
 
     x = array({1.0f, NAN, 3.0f, 4.0f, 5.0f, 6.0f});
     CHECK(isnan(max(x).item<float>()));
-
   }
 
   // Test logsumexp