From a42d22e1d12ab9c6bb3de5c17d62ab7aece90828 Mon Sep 17 00:00:00 2001 From: v-chen_data Date: Mon, 16 Sep 2024 15:56:07 -0700 Subject: [PATCH] precommit --- llmfoundry/command_utils/eval.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/llmfoundry/command_utils/eval.py b/llmfoundry/command_utils/eval.py index dbdc448f31..5c9cb900fe 100644 --- a/llmfoundry/command_utils/eval.py +++ b/llmfoundry/command_utils/eval.py @@ -12,7 +12,7 @@ from composer.core import Callback from composer.loggers.logger_destination import LoggerDestination from composer.trainer import Trainer -from composer.utils import dist, get_device, reproducibility, parallelism +from composer.utils import dist, get_device, parallelism, reproducibility from omegaconf import DictConfig from omegaconf import OmegaConf as om @@ -70,7 +70,9 @@ def evaluate_model( ): throw_deprecation_warning = False if parallelism_config: - deprecated_fsdp_args = list(parallelism.FSDPConfig.__annotations__.keys()) + deprecated_fsdp_args = list( + parallelism.FSDPConfig.__annotations__.keys() + ) for deprecated_arg in deprecated_fsdp_args: if deprecated_arg in parallelism_config: throw_deprecation_warning = True