[BUG] Dictionary key is unintentionally modified by Flyte when using custom python objects to configure tasks

[fg91]

Describe the bug

When configuring Flyte tasks with Python dataclasses and when using dictionaries within those dataclasses, flyte, under certain circumstances, replaces the name of the key with the name of the dictionary.

Expected behavior

Flyte should not modify the arguments.

Additional context to reproduce

Let's create a custom python dataclass to configure our task/workflow:

@dataclass_json
@dataclass
class Hyperparams(object):
    param_dict: Dict[str, str]  # Used e.g. to pass additional kwargs
    # ... more hyper parameters

@task
def train(
    hparams: Hyperparams,
) -> str:
    print(hparams)


@workflow
def workflow(
    hparams: Hyperparams = Hyperparams(
        param_dict={"foo": "bar"}
    ),

) -> None:
    train(
        hparams=hparams,
    )

Screenshots

In the Flyteconsole, the name of the dict is displayed instead of the key:

The workflow is started with the correct arguments ...

... unless, in Flyteconsole, one puts the cursor into the param_dict (string) field:

I used image: cr.flyte.org/flyteorg/flyteconsole-release:v1.2.0.

Are you sure this issue hasn't been raised already?

• Yes

Have you read the Code of Conduct?

• Yes

[wild-endeavor]

@jsonporter could you take a look at this please? it looks to be localized to the UI.

[jsonporter]

Closed with flyteorg/flyteconsole#692