fix test

Author: awaelchli

tests/tests_pytorch/utilities/test_deepspeed_model_summary.py

@@ -22,7 +22,7 @@
 
 
 @RunIf(min_cuda_gpus=2, deepspeed=True, standalone=True)
-def test_deepspeed_summary(tmpdir):
+def test_deepspeed_summary(tmp_path):
     """Test to ensure that the summary contains the correct values when stage 3 is enabled and that the trainer enables
     the `DeepSpeedSummary` when DeepSpeed is used."""
 
@@ -37,12 +37,12 @@ def on_fit_start(self, trainer: "pl.Trainer", pl_module: "pl.LightningModule") -
 
             # check the additional params per device
             summary_data = model_summary._get_summary_data()
-            params_per_device = summary_data[-1][-1]
+            params_per_device = summary_data[4][-1]
             assert int(params_per_device[0]) == (model_summary.total_parameters // 2)
 
     trainer = Trainer(
         strategy=DeepSpeedStrategy(stage=3),
-        default_root_dir=tmpdir,
+        default_root_dir=tmp_path,
         accelerator="gpu",
         fast_dev_run=True,
         devices=2,