add trailing comma

mosaicml · Feb 26, 2024 · 954c21a · 954c21a
1 parent 54171a5
commit 954c21a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/trainer/test_fsdp.py b/tests/trainer/test_fsdp.py
@@ -220,8 +220,13 @@ def test_fsdp_process_group(world_size: int):
 @pytest.mark.gpu
 @world_size(2)
 @pytest.mark.skipif(version.parse(torch.__version__) < version.parse('2.2.0'), reason='Device mesh requires Torch 2.2')
-@pytest.mark.parametrize('sharding_strategy',
-                         ['NO_SHARD', 'SHARD_GRAD_OP', 'FULL_SHARD', 'HYBRID_SHARD', '_HYBRID_SHARD_ZERO2'])
+@pytest.mark.parametrize('sharding_strategy', [
+    'NO_SHARD',
+    'SHARD_GRAD_OP',
+    'FULL_SHARD',
+    'HYBRID_SHARD',
+    '_HYBRID_SHARD_ZERO2',
+])
 @pytest.mark.parametrize('device_mesh', [[2], [1, 2]])
 def test_wrong_size_device_mesh_error(world_size: int, sharding_strategy: str, device_mesh: list[int]):
     context = contextlib.nullcontext()