From 76305f9f4facee64e477c5c4ed8665c2c0551cc3 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 26 Feb 2024 15:05:27 -0500 Subject: [PATCH] hotfix to exclude_unset from pydantic config when converting back to a dict --- src/axolotl/utils/config/__init__.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/src/axolotl/utils/config/__init__.py b/src/axolotl/utils/config/__init__.py index 6635ff8e2a..d289054f49 100644 --- a/src/axolotl/utils/config/__init__.py +++ b/src/axolotl/utils/config/__init__.py @@ -199,9 +199,15 @@ def normalize_cfg_datasets(cfg): def validate_config(cfg: DictDefault, capabilities: Optional[dict] = None): if capabilities: return DictDefault( - dict(AxolotlConfigWCapabilities(**cfg.to_dict(), capabilities=capabilities)) + dict( + AxolotlConfigWCapabilities( + **cfg.to_dict(), capabilities=capabilities + ).model_dump(exclude_unset=True) + ) ) - return DictDefault(dict(AxolotlInputConfig(**cfg.to_dict()))) + return DictDefault( + dict(AxolotlInputConfig(**cfg.to_dict()).model_dump(exclude_unset=True)) + ) def legacy_validate_config(cfg):