From 269c5436ead4c905b4d3e2e5c78a6f626216059d Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 26 Feb 2024 15:06:25 -0500 Subject: [PATCH] hotfix to exclude_unset from pydantic config when converting back to a dict (#1334) --- 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 6635ff8e2..d289054f4 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):