From ceb6591cb811a704d119874a2dc591b77144eaef Mon Sep 17 00:00:00 2001 From: v-chen_data Date: Sat, 30 Nov 2024 23:41:04 -0500 Subject: [PATCH] precommit --- tests/test_events.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_events.py b/tests/test_events.py index b84c90327f..396e5c2fa9 100644 --- a/tests/test_events.py +++ b/tests/test_events.py @@ -31,7 +31,7 @@ def get_trainer(self, precision='fp32', max_duration='1ep', **kwargs): train_dataset = RandomClassificationDataset() eval_dataset = RandomClassificationDataset() - train_batch_size = 2 + train_batch_size = 2 evaluator1 = DataLoader( dataset=eval_dataset, @@ -99,7 +99,7 @@ def get_trainer(self, precision='fp32', max_duration='1ep', **kwargs): ), ], ) - @pytest.mark.parametrize('save_interval', ['1ep']) + @pytest.mark.parametrize('save_interval', ['1ep', '1ba']) def test_event_calls(self, world_size, device, deepspeed_zero_stage, use_fsdp, precision, save_interval): save_interval = Time.from_timestring(save_interval)