Skip to content

Commit 707ba66

Browse files
committed
Fix edge case of n=0 in seed generator
1 parent 445f57d commit 707ba66

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

src/imitation/util/util.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,9 @@ def make_seeds(
177177
Returns:
178178
A list of n random seeds.
179179
"""
180-
seeds: List[int] = rng.integers(0, (1 << 31) - 1, (n or 1,)).tolist()
180+
seeds: List[int] = rng.integers(
181+
0, (1 << 31) - 1, (n if n is not None else 1,)
182+
).tolist()
181183
if n is None:
182184
return seeds[0]
183185
else:

0 commit comments

Comments
 (0)