Also another question, when trying to update the configuration as following

lstm_config = AutoLSTM.get_default_config(h = h, backend="optuna")
def config_lstm(trial):
    config = {**lstm_config(trial)}
    config.update({
                   "input_size": trial.suggest_int("input_size", 2, 18),
                   })
    return config
lstm_config

modelLSTM = AutoLSTM(h=h,
                     config=config_lstm,
                     backend='optuna',
                     loss=MAE(),
                     num_samples=3)

During fitting I get the following error raise: ValueError("Cannot set different distribution kind to the same parameter name.") ValueError: Cannot set different distribution kind to the same parameter name. [W 2025-03-18 122842,864] Trial 0 failed with value None.

Could you try with this:

lstm_config = AutoLSTM.get_default_config(h=h, backend="optuna")

def config_lstm(trial):
    config = lstm_config.copy()
    config.update({
        "input_size": trial.suggest_int("input_size", 2, 18),
    })
    return config