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..f6dae3a8 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,11 +31,12 @@ 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": + case "predictor_corrector" | "adaptive_corrector": sampling_parameters = PredictorCorrectorSamplingParameters( **sampling_parameter_dictionary )