Skip to content

Commit

Permalink
Add transforms to logged config (#1428)
Browse files Browse the repository at this point in the history
Co-authored-by: Saaketh Narayan <[email protected]>
  • Loading branch information
b-chu and snarayan21 authored Aug 5, 2024
1 parent 32803b7 commit 6dcc18a
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions llmfoundry/utils/config_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -332,15 +332,15 @@ def make_dataclass_and_log_config(
'icl_tasks must be specified in the config',
)

# Create copy of config for logging
logged_cfg: Dict[str, Any] = copy.deepcopy(unstructured_config)

# Apply transforms to the unstructured config before constructing dataclass
unstructured_config = apply_transforms_to_config(
unstructured_config,
transforms,
)

# Create copy of config for logging
logged_cfg: Dict[str, Any] = copy.deepcopy(unstructured_config)

arg_config_keys = set(unstructured_config.keys())
extraneous_keys = set.difference(arg_config_keys, dataclass_fields)

Expand Down

0 comments on commit 6dcc18a

Please sign in to comment.