fix deepspeed assertion

justusschock · justusschock · commit 9ec3639bb96e · 2025-11-17T13:10:18.000Z
diff --git a/tests/tests_fabric/strategies/test_deepspeed_integration.py b/tests/tests_fabric/strategies/test_deepspeed_integration.py
@@ -164,7 +164,7 @@ def test_deepspeed_custom_precision_params():
         devices=1,
     )
     fabric.launch()
-    assert fabric._strategy._config_initialized
+    assert fabric._strategy.deepspeed_impl._config_initialized
     assert fabric._strategy.config["fp16"]["loss_scale"] == 10
     assert fabric._strategy.config["fp16"]["initial_scale_power"] == 11
     assert fabric._strategy.config["fp16"]["loss_scale_window"] == 12

Original file line number	Diff line number	Diff line change
`@@ -164,7 +164,7 @@ def test_deepspeed_custom_precision_params():`
`164`	`164`	`devices=1,`
`165`	`165`	`)`
`166`	`166`	`fabric.launch()`
`167`		`- assert fabric._strategy._config_initialized`
	`167`	`+ assert fabric._strategy.deepspeed_impl._config_initialized`
`168`	`168`	`assert fabric._strategy.config["fp16"]["loss_scale"] == 10`
`169`	`169`	`assert fabric._strategy.config["fp16"]["initial_scale_power"] == 11`
`170`	`170`	`assert fabric._strategy.config["fp16"]["loss_scale_window"] == 12`