integrations.config

integrations.config

Module to handle merging the plugins’ input arguments with the base configurations.

This was moved here to prevent circular imports.

Functions

Name Description
merge_input_args Merges input arguments from registered plugins with the base configurations.
merge_training_args Merges training arguments from registered plugins with the base TrainingArguments.

merge_input_args

integrations.config.merge_input_args()

Merges input arguments from registered plugins with the base configurations.

This function retrieves the input arguments from registered plugins using the PluginManager. It then dynamically creates new classes, AxolotlConfigWCapabilities and AxolotlInputConfig, that inherit from the base configurations and include the input arguments from the plugins.

Returns: tuple: A tuple containing the newly created classes, AxolotlConfigWCapabilities and AxolotlInputConfig.

merge_training_args

integrations.config.merge_training_args()

Merges training arguments from registered plugins with the base TrainingArguments.

This function retrieves the training arguments from registered plugins using the PluginManager. It then dynamically creates new classes, AxolotlTrainingMixins, that inherit from the base configurations and include the training arguments from the plugins.

Returns: tuple: A tuple containing the newly created classes, AxolotlTrainingMixins.