cli.shared.types.defense package

cli.shared.types.defense.adversarial_training module

class cli.shared.types.defense.adversarial_training.ATCliConfigType(training: TrainingCliConfigType, adversarial_training: AdversarialTrainingConfig)

Bases: object

This class is used as a type hint for the Adversarial Training CLI configuration.

adversarial_training: AdversarialTrainingConfig

training: TrainingCliConfigType

class cli.shared.types.defense.adversarial_training.AdversarialTrainingConfig(models: List[ModelWithConfigDict], attacks: List[Dict[str, str]])

Bases: object

This class is used to store the configuration of the adversarial training.

attacks: List[Dict[str, str]]

models: List[ModelWithConfigDict]

class cli.shared.types.defense.adversarial_training.ModelWithConfigDict(config: str)

Bases: TypedDict

This class is used as a type hint for a model with its configuration.

config: str