Fix tests

Author: PetrochukM

torchnlp/samplers/bucket_batch_sampler.py

@@ -1,5 +1,6 @@
 import heapq
 import pickle
+import math
 
 from torch.utils.data.sampler import BatchSampler
 from torch.utils.data.sampler import RandomSampler
@@ -114,4 +115,4 @@ def __len__(self):
         if self.drop_last:
             return len(self.data) // self.batch_size
         else:
-            return len(self.data) // self.batch_size + 1
+            return math.ceil(len(self.data) / self.batch_size)

torchnlp/samplers/noisy_sorted_batch_sampler.py

@@ -74,4 +74,4 @@ def __iter__(self):
             random.shuffle(batches)
         if self.last_batch_first:
             batches.insert(0, last_batch)
-        return iter(batches)
+        return iter(batches)