Skip to content

Commit 0389f9e

Browse files
marigold_ensemble integer fix (#389)
Code strengthening
1 parent c7e6a56 commit 0389f9e

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/backbone.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,10 @@ def gather_ops():
4242
c = get_cmd_opt(s, None)
4343
if c is not None:
4444
ops[s] = c
45+
# sanitize for integers.
46+
for s in ['marigold_ensembles', 'marigold_steps']:
47+
if s in ops:
48+
ops[s] = int(ops[s])
4549
return ops
4650

4751

0 commit comments

Comments
 (0)