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 28c20f59..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 @@ -36,11 +36,7 @@ def load_sampling_parameters( ], "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": + case "predictor_corrector" | "adaptive_corrector": sampling_parameters = PredictorCorrectorSamplingParameters( **sampling_parameter_dictionary )