Update sampler.py

constantinpape · Jul 11, 2024 · 380b2a5 · 380b2a5
1 parent 1c9dbe8
commit 380b2a5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_em/data/sampler.py b/torch_em/data/sampler.py
@@ -1,5 +1,5 @@
 import numpy as np
-from typing import List
+from typing import List, Optional
 
 
 class MinForegroundSampler:
@@ -86,7 +86,7 @@ def __init__(
  self,
  min_num_instances: int = 2,
  p_reject: float = 1.0,
- min_size: int = None
+ min_size: Optional[int] = None
  ):
  self.min_num_instances = min_num_instances
  self.p_reject = p_reject