Merge pull request #214 from Modalities/fix/linting

Fix linting
Modalities · Aug 12, 2024 · 44bfe4c · 44bfe4c
2 parents 9829679 + bbff3c8
commit 44bfe4c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/utils/test_number_conversion.py b/tests/utils/test_number_conversion.py
@@ -5,7 +5,7 @@
 
 @pytest.mark.parametrize(
     "num_ranks,global_num_samples,local_micro_batch_size,expected",
-    [(2, 100, 10, 5), (2, 110, 10, 5), (4, 100, 10, 2), (4, 100, 5, 5)]
+    [(2, 100, 10, 5), (2, 110, 10, 5), (4, 100, 10, 2), (4, 100, 5, 5)],
 )
 def test_get_local_num_batches_from_num_samples(
     num_ranks: int, global_num_samples: int, local_micro_batch_size: int, expected: int