bug fix: add int type for rng seed (#192)

* bug fix: add int type for rng seed * fix: update the version for this bug fix * update entry point test with rng seed * Fix: remove double line Co-authored-by: Sander Roet <[email protected]> --------- Co-authored-by: Sander Roet <[email protected]>
SBC-Utrecht · Jul 5, 2024 · 2e89746 · 2e89746
1 parent 609d6fb
commit 2e89746
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pytom-match-pick"
-version = "0.7.0"
+version = "0.7.1"
 description = "PyTOM's GPU template matching module as an independent package"
 readme = "README.md"
 license = {file = "LICENSE"}

diff --git a/src/pytom_tm/entry_points.py b/src/pytom_tm/entry_points.py
@@ -786,6 +786,7 @@ def match_template(argv=None):
     )
     additional_group.add_argument(
         "--rng-seed",
+        type=int,
         action=LargerThanZero,
         default=int.from_bytes(urandom(8)),
         required=False,

diff --git a/tests/test_entry_points.py b/tests/test_entry_points.py
@@ -158,6 +158,9 @@ def start(arg_dict):  # simplify run
         arguments = defaults.copy()
         arguments["-r"] = ""
         start(arguments)
+        # test if we can set the rng seed, see issue #194
+        arguments["--rng-seed"] = "42"
+        start(arguments)
 
         # test debug files
         arguments = defaults.copy()