Skip to content

Commit 00be545

Browse files
committed
fix: handle imgaug without seed function
1 parent 96add57 commit 00be545

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

lantern/set_seeds.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ def set_seeds(seed):
1111
try:
1212
import imgaug
1313

14-
imgaug.seed(seed + 4)
14+
if hasattr(imgaug, "seed"):
15+
imgaug.seed(seed + 4)
1516
except ImportError:
1617
pass

0 commit comments

Comments
 (0)