small changes

Benny-Nottonson · Mar 4, 2024 · b0949fb · b0949fb
1 parent 6b5c99d
commit b0949fb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/datasets/mnist/__init__.mojo b/datasets/mnist/__init__.mojo
@@ -43,4 +43,10 @@ struct MNist:
                 self.test_images[i][j - 1] = atol(line[j])
 
         info("MNIST dataset loaded.\n")
-        print("There are ", train_size, " training samples and ", test_size, " test samples.\n")
+        print(
+            "There are ",
+            train_size,
+            " training samples and ",
+            test_size,
+            " test samples.\n",
+        )
diff --git a/voodoo/core/tensor.mojo b/voodoo/core/tensor.mojo
@@ -1,4 +1,5 @@
 from tensor import TensorShape
+from collections import Optional
 
 from voodoo.autograd import Node, Graph
 from voodoo.utils import (