Add `replace_subgroups` function

[zhiruiluo]

This PR is to add the replace_selections function that replaces some values in a dataclass and replaces dataclass type in nested union of dataclasses or subgroups.

simple_parsing.replace_selections function support replacingeplaces dataclass type in nested union of dataclasses or subgroups in addition to simple_parsing.replace #212.

• signature: replace_selections(obj: DataclassT, changes_dict: dict[str, Any] | None = None, selections: dict[str, Key | DataclassT] | None = None, **changes) -> DataclassT:
  • Compared to simple_replace.replace, this calls replace_selected_dataclass before calling simple_parsing.replace.
• helper function replace_selected_dataclass
  • This function replaces the dataclass of subgroups, union, and optional union. The selections is in flat format, e.g. {"ab_or_cd": 'cd', "ab_or_cd.c_or_d": 'd'}
  • The values of selections can be Key of subgroups, dataclass type, and dataclass instance.

Open question:

• Do we combine the simple_parsing.replace_selections and simple_parsing.replace?

Examples

    >>> import dataclasses
    >>> from simple_parsing import replace_selections, subgroups
    >>> from typing import Union
    >>> @dataclasses.dataclass
    ... class A:
    ...     a: int = 0
    >>> @dataclasses.dataclass
    ... class B:
    ...     b: str = "b"
    >>> @dataclasses.dataclass
    ... class Config:
    ...     a_or_b: Union[A, B] = subgroups({'a': A, 'b': B}, default_factory=A)
    ...     a_or_b_union: Union[A, B] = dataclasses.field(default_factory=A)
    ...     a_optional: Union[A, None] = None

    >>> base_config = Config(a_or_b=A(a=1))
    # Replace subgroups field by subgroup `Key`, dataclass type, or dataclass instance
    >>> replace_selections(base_config, {"a_or_b.b": "bob"}, {"a_or_b": "b"})
    Config(a_or_b=B(b='bob'), a_or_b_union=A(a=0), a_optional=None)
    >>> replace_selections(base_config, {"a_or_b.b": "bob"}, {"a_or_b": B})
    Config(a_or_b=B(b='bob'), a_or_b_union=A(a=0), a_optional=None)
    >>> replace_selections(base_config, {}, {"a_or_b": B(b="bob")})
    Config(a_or_b=B(b='bob'), a_or_b_union=A(a=0), a_optional=None)

    # Replace union of dataclasses and optional dataclass
    >>> replace_selections(base_config, {"a_or_b_union.b": "bob"}, {"a_or_b_union": B})
    Config(a_or_b=A(a=1), a_or_b_union=B(b='bob'), a_optional=None)
    >>> replace_selections(base_config, {"a_optional.a": 10}, {"a_optional": A})
    Config(a_or_b=A(a=1), a_or_b_union=A(a=0), a_optional=A(a=10))

[zhiruiluo]

WIP

[zhiruiluo]

Hi @lebrice, Sorry for letting you wait too long.

Here is the updating:

• keep only replace_subgroups function
• simplify test cases
• move replace_subgroups function into replace.py file
• move the unflatten function out of utils.py

[zhiruiluo]

Hi @lebrice, Here are some updates so far:

• Fixed the bug in unflatten_selection_dict when no overlapping of keys and added a recursive flag.
• reordered some logics for readability
• add test case for the frozen dataclass to work

I would appreciate any further feedback or suggestions for improvement that you may have. Thank you.