From c06fc7f485f81ad63f5cd0a176927df7ee3bce6c Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Wed, 17 Dec 2025 20:12:32 +0100 Subject: [PATCH] use argparser for project selection --- src/rbm/rbm.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/src/rbm/rbm.py b/src/rbm/rbm.py index e099c9d..bf0f20a 100644 --- a/src/rbm/rbm.py +++ b/src/rbm/rbm.py @@ -1,6 +1,7 @@ import os.path import numpy as np import cv2 as cv +from argparse import ArgumentParser from stack_factory import StackFactory from status import Status from stack_deep import StackDeep @@ -90,8 +91,15 @@ def main(prj_name: str = "test"): stack.state_save() if __name__ == "__main__": -# main("norb_small_16h_v2") - main("mnist_2") + ap = ArgumentParser() + ap.add_argument("name", type=str, + default='default', + help="Name of project") + + args = ap.parse_args() + var_args = vars(args) + + main(var_args["name"]) cv.destroyAllWindows() print("Test: [passed]")