modified: examples/tensorops/mnist.py

	modified:   tensorops/backend.py
	modified:   tensorops/loss.py
	modified:   tensorops/mlx_runtime.py
	modified:   tensorops/model.py
	modified:   tensorops/optim.py
	modified:   tensorops/src/kernel.cl
	modified:   tensorops/src/kernel.rs
	modified:   tensorops/src/runtime.rs
	modified:   tensorops/tensor.py
	modified:   tensorops/utils/models.py

Author: andreaslam2

examples/tensorops/mnist.py

@@ -40,9 +40,9 @@ def extract_images(filepath):
     with gzip.open(filepath, "rb") as f:
         header = f.read(16)
         magic_number, num_images, rows, cols = struct.unpack(">IIII", header)
-        assert magic_number == 2051, (
-            f"Invalid magic number {magic_number} in image file."
-        )
+        assert (
+            magic_number == 2051
+        ), f"Invalid magic number {magic_number} in image file."
 
         images = []
         for _ in range(num_images):
@@ -58,9 +58,9 @@ def extract_labels(filepath):
     with gzip.open(filepath, "rb") as f:
         header = f.read(8)
         magic_number, num_labels = struct.unpack(">II", header)
-        assert magic_number == 2049, (
-            f"Invalid magic number {magic_number} in label file."
-        )
+        assert (
+            magic_number == 2049
+        ), f"Invalid magic number {magic_number} in label file."
 
         labels = list(f.read(num_labels))
         return labels