Update test_deepspeed.py

amorehead · web-flow · commit c37ea0015570 · 2025-03-19T09:41:52.000-05:00
diff --git a/tests/tests_fabric/strategies/test_deepspeed.py b/tests/tests_fabric/strategies/test_deepspeed.py
@@ -137,6 +137,7 @@ def test_deepspeed_setup_module(init_mock):
         model=model,
         model_parameters=ANY,
         optimizer=None,
+        lr_scheduler=None,
         dist_init_required=False,
     )
 

Original file line number	Diff line number	Diff line change
`@@ -137,6 +137,7 @@ def test_deepspeed_setup_module(init_mock):`
`137`	`137`	`model=model,`
`138`	`138`	`model_parameters=ANY,`
`139`	`139`	`optimizer=None,`
	`140`	`+ lr_scheduler=None,`
`140`	`141`	`dist_init_required=False,`
`141`	`142`	`)`
`142`	`143`