fix

analysiscenter · Dec 26, 2023 · b49070d · b49070d
1 parent 7402cb6
commit b49070d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/batchflow/sampler.py b/batchflow/sampler.py
@@ -343,7 +343,8 @@ def sample(self, size):
 class WeightedSampler(Sampler):
     """ Class for implementing `&` (weighting) operation on a number and a `Sampler` instance.
     """
-    def __init__(self, base, weight):
+    def __init__(self, base, weight, *args, **kwargs):
+        super().__init__(*args, **kwargs)
         self.bases = [base]
         self.weight = weight