Skip to content

Commit c36e00a

Browse files
author
SeanNaren
committed
Fix error in refactor
1 parent 78f1eb4 commit c36e00a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/accelerators/test_accelerator_connector.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,8 @@
2626
from pytorch_lightning.plugins import (
2727
DDP2Plugin,
2828
DDPPlugin,
29+
DDPShardedPlugin,
2930
DDPSpawnPlugin,
30-
FullShardedPlugin,
3131
PrecisionPlugin,
3232
SingleDevicePlugin,
3333
)
@@ -396,7 +396,7 @@ def test_plugin_accelerator_choice(accelerator, plugin):
396396
Ensure that when a plugin and accelerator is passed in, that the plugin takes precedent.
397397
"""
398398
trainer = Trainer(accelerator=accelerator, plugins=plugin, num_processes=2)
399-
assert isinstance(trainer.accelerator.training_type_plugin, FullShardedPlugin)
399+
assert isinstance(trainer.accelerator.training_type_plugin, DDPShardedPlugin)
400400

401401
trainer = Trainer(plugins=plugin, num_processes=2)
402-
assert isinstance(trainer.accelerator.training_type_plugin, FullShardedPlugin)
402+
assert isinstance(trainer.accelerator.training_type_plugin, DDPShardedPlugin)

0 commit comments

Comments
 (0)