Updating tests for rint version

ancestor-mithril · Jun 5, 2024 · 2ce3162 · 2ce3162
1 parent 019aabe
commit 2ce3162
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_ChainedBSScheduler.py b/tests/test_ChainedBSScheduler.py
@@ -24,7 +24,7 @@ def test_dataloader_lengths(self):
         n_epochs = 10
 
         epoch_lengths = simulate_n_epochs(dataloader, scheduler, n_epochs)
-        expected_batch_sizes = [100, 110, 121, 133, 14, 16, 18, 20, 22, 24]
+        expected_batch_sizes = [100, 110, 121, 133, 14, 16, 17, 19, 21, 23]
         expected_lengths = self.compute_epoch_lengths(expected_batch_sizes, len(self.dataset), drop_last=False)
         self.assertEqual(epoch_lengths, expected_lengths)
 
@@ -37,7 +37,7 @@ def test_dataloader_batch_size(self):
         n_epochs = 10
 
         batch_sizes = get_batch_sizes_across_epochs(dataloader, scheduler, n_epochs)
-        expected_batch_sizes = [100, 110, 121, 133, 14, 16, 18, 20, 22, 24]
+        expected_batch_sizes = [100, 110, 121, 133, 14, 16, 17, 19, 21, 23]
 
         self.assertEqual(batch_sizes, expected_batch_sizes)