From bbff3c8a5d4c1009d74cff928c3dde896da2b048 Mon Sep 17 00:00:00 2001 From: Felix Stollenwerk Date: Thu, 8 Aug 2024 19:44:57 +0200 Subject: [PATCH] chore: fix linting --- tests/utils/test_number_conversion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/utils/test_number_conversion.py b/tests/utils/test_number_conversion.py index 1e06ba70..9200ea73 100644 --- 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