From 694f2e2c12efbda81a1aa4b4b486767264116a2f Mon Sep 17 00:00:00 2001 From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com> Date: Mon, 11 Dec 2023 16:15:23 +0530 Subject: [PATCH] fix the failing test (#2237) --- tests/deepspeed/test_deepspeed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/deepspeed/test_deepspeed.py b/tests/deepspeed/test_deepspeed.py index 9b669827b84..954df2e764d 100644 --- a/tests/deepspeed/test_deepspeed.py +++ b/tests/deepspeed/test_deepspeed.py @@ -338,7 +338,7 @@ def test_prepare_deepspeed(self, optim_type, scheduler_type): model, optimizer, lr_scheduler = accelerator.prepare(model, optimizer, lr_scheduler) self.assertTrue( "When using DeepSpeed `accelerate.prepare()` requires you to pass at least one of training or evaluation dataloaders " - "or alternatively set an integer value in `train_micro_batch_size_per_gpu` in the deepspeed config file" + "or alternatively set an integer value in `train_micro_batch_size_per_gpu` in the deepspeed config file " "or assign integer value to `AcceleratorState().deepspeed_plugin.deepspeed_config['train_micro_batch_size_per_gpu']`." in str(cm.exception) )