adress comments

Lightning-AI · Feb 28, 2022 · a90a8a4 · a90a8a4
1 parent e8418c2
commit a90a8a4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pytorch_lightning/trainer/connectors/accelerator_connector.py b/pytorch_lightning/trainer/connectors/accelerator_connector.py
@@ -304,17 +304,17 @@ def _check_config_and_set_final_flags(
                         f"Passing {plugin} `strategy` to the `plugins` flag in Trainer has been deprecated"
                         f" in v1.5 and will be removed in v1.7. Use `Trainer(strategy={plugin})` instead."
                     )
-                    plugins_flags_types_list.append("Strategy")
+                    plugins_flags_types_list.append(Strategy.__name__)
 
                 elif isinstance(plugin, PrecisionPlugin):
                     self._precision_plugin_flag = plugin
-                    plugins_flags_types_list.append("PrecisionPlugin")
+                    plugins_flags_types_list.append(PrecisionPlugin.__name__)
                 elif isinstance(plugin, CheckpointIO):
                     self.checkpoint_io = plugin
-                    plugins_flags_types_list.append("CheckpointIO")
+                    plugins_flags_types_list.append(CheckpointIO.__name__)
                 elif isinstance(plugin, ClusterEnvironment):
                     self._cluster_environment_flag = plugin
-                    plugins_flags_types_list.append("ClusterEnvironment")
+                    plugins_flags_types_list.append(ClusterEnvironment.__name__)
                 else:
                     raise MisconfigurationException(
                         f"Found invalid type for plugin {plugin}. Expected PrecisionPlugin, "