diff --git a/src/diffusion_for_multi_scale_molecular_dynamics/generators/load_sampling_parameters.py b/src/diffusion_for_multi_scale_molecular_dynamics/generators/load_sampling_parameters.py index 99f1ccfe..28c20f59 100644 --- a/src/diffusion_for_multi_scale_molecular_dynamics/generators/load_sampling_parameters.py +++ b/src/diffusion_for_multi_scale_molecular_dynamics/generators/load_sampling_parameters.py @@ -31,14 +31,19 @@ def load_sampling_parameters( assert algorithm in [ "ode", "sde", + "adaptive_corrector", "predictor_corrector", - ], "Unknown algorithm. Possible choices are 'ode', 'sde' and 'predictor_corrector'" + ], "Unknown algorithm. Possible choices are 'ode', 'sde', 'predictor_corrector' and 'adaptive_corrector'" match algorithm: case "predictor_corrector": sampling_parameters = PredictorCorrectorSamplingParameters( **sampling_parameter_dictionary ) + case "adaptive_corrector": + sampling_parameters = PredictorCorrectorSamplingParameters( + **sampling_parameter_dictionary + ) case "ode": sampling_parameters = ODESamplingParameters(**sampling_parameter_dictionary) case "sde":