Common shared CLI flags for different scripts? about tyro

[mainrs]

I didn't think about inheritance, sorry! Thanks for helping me out!

[brentyi]

Hi @mainrs!

Here are some runnable options for getting --seed INT and --flag1 INT. Do any of them help?

(1) Standard dataclass inheritance might be the easiest:

from dataclasses import dataclass
import tyro

@dataclass
class CommonConfig:
    seed: int = 0

@dataclass
class CustomConfig(CommonConfig):
    flag1: int = 1

tyro.cli(CustomConfig)

(2) tyro.conf.OmitArgPrefixes will flatten all of the arguments:

from dataclasses import dataclass
import tyro

@dataclass
class CommonConfig:
    seed: int = 0

@dataclass
class CustomConfig:
    common: CommonConfig
    flag1: int = 1

tyro.cli(tyro.conf.OmitArgPrefixes[CustomConfig])

(3) You can "erase" the name of the common field, which will remove the common. prefix. This is the most cryptic option (and frankly not well-documented), but it's in the tyro tests:

from dataclasses import dataclass
import tyro
from typing import Annotated

@dataclass
class CommonConfig:
    seed: int = 0

@dataclass
class CustomConfig:
    common: Annotated[CommonConfig, tyro.conf.arg(name="")]
    flag1: int = 1

tyro.cli(CustomConfig)