fix: take could result in more samples than max

Author: samedii

datastream/samplers.py

@@ -24,7 +24,7 @@ def __init__(self, length, proportion=1.0, replacement=False):
             replacement=replacement,
             sampler=torch.utils.data.WeightedRandomSampler(
                 torch.ones(length).double(),
-                num_samples=max(int(length * proportion), 1),
+                num_samples=int(max(1, min(length, length * proportion))),
                 replacement=replacement,
             )
         )