From 790e25e2fd9c723991cc962655e3c5eee8255b62 Mon Sep 17 00:00:00 2001 From: Saaketh Date: Tue, 3 Dec 2024 11:32:40 -0500 Subject: [PATCH] yo --- tests/test_events.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_events.py b/tests/test_events.py index 80651e75a2..d6407df2d7 100644 --- a/tests/test_events.py +++ b/tests/test_events.py @@ -80,7 +80,7 @@ def get_trainer(self, precision='fp32', max_duration='1ep', save_interval='1ep', @pytest.mark.parametrize( 'device,use_fsdp,precision', [ - pytest.param('cpu', None, False, 'fp32', id='cpu-ddp'), + pytest.param('cpu', False, 'fp32', id='cpu-ddp'), # TODO: Remove filterwarnings after FSDP remove deprecated code pytest.param( 'gpu',