diff --git a/src/lightning/pytorch/trainer/connectors/accelerator_connector.py b/src/lightning/pytorch/trainer/connectors/accelerator_connector.py index 7af8f13fce38a..b892fcc3290e4 100644 --- a/src/lightning/pytorch/trainer/connectors/accelerator_connector.py +++ b/src/lightning/pytorch/trainer/connectors/accelerator_connector.py @@ -15,6 +15,7 @@ import logging import os from collections import Counter +from collections.abc import Iterable from typing import Literal, Optional, Union import torch @@ -182,7 +183,7 @@ def _check_config_and_set_final_flags( """ if plugins is not None: - plugins = [plugins] if not isinstance(plugins, list) else plugins + plugins = [plugins] if not isinstance(plugins, Iterable) else plugins if isinstance(strategy, str): strategy = strategy.lower()