InitVar

michaelbyun · Mar 22, 2023 · d304ab3 · d304ab3
1 parent a80369e
commit d304ab3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/elk/extraction/balanced_sampler.py b/elk/extraction/balanced_sampler.py
@@ -26,10 +26,10 @@ class BalancedSampler(TorchIterableDataset):
 
  dataset: IterableDataset
  label_counts: np.ndarray = field(default_factory=lambda: np.zeros(2))
- seed: int = 42
+ seed: InitVar[int] = 42
 
- def __post_init__(self):
- self.rng = np.random.default_rng(self.seed)
+ def __post_init__(self, seed: int):
+ self.rng = np.random.default_rng(seed)
 
  def __iter__(self):
  for sample in self.dataset: