From af9a0c70b21bcaffeb2bcdd7eb85500c5be9ec78 Mon Sep 17 00:00:00 2001
From: Lenoctambule <106790775+lenoctambule@users.noreply.github.com>
Date: Fri, 27 Mar 2026 06:06:32 +0100
Subject: [PATCH] feat: handle args for mnist test

---
 mnist_test.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/mnist_test.py b/mnist_test.py
index daae9e5..7b62a39 100644
--- a/mnist_test.py
+++ b/mnist_test.py
@@ -53,4 +53,14 @@ def mnist_embed(
 
 
 if __name__ == "__main__":
-    mnist_embed(10, 1000, 5)
+    import argparse
+    import sys
+
+    options = "b:e:p:"
+    parser = argparse.ArgumentParser()
+    parser.add_argument('-b', type=int, nargs='+', default=50)
+    parser.add_argument('-e', type=int, nargs='+', default=1000)
+    parser.add_argument('-p', type=int, nargs='+', default=5)
+    args = parser.parse_args(sys.argv[1:])
+
+    mnist_embed(args.b, args.e, args.p)