Skip to content

Commit 46e467b

Browse files
committed
ENH: --random-seed cmdline option
1 parent 8fbe601 commit 46e467b

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

bin/heudiconv

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2165,6 +2165,8 @@ def get_parser():
21652165
"not both")
21662166
parser.add_argument('--minmeta', dest='minmeta', action='store_true',
21672167
help="Exclude dcmstack's meta information in sidecar jsons")
2168+
parser.add_argument('--random-seed', type=int, default=None,
2169+
help='Random seed to initialize RNG')
21682170
return parser
21692171

21702172

@@ -2173,6 +2175,12 @@ def main(argv=None):
21732175
"""
21742176
parser = get_parser()
21752177
args = parser.parse_args(argv)
2178+
# To be done asap so anything random is deterministic
2179+
if args.random_seed is not None:
2180+
import random
2181+
random.seed(args.random_seed)
2182+
import numpy
2183+
numpy.random.seed(args.random_seed)
21762184

21772185
# TODO: assign distribution-restrictions=sensitive
21782186

0 commit comments

Comments
 (0)