Better match case.

mila-iqia · Dec 3, 2024 · 7919c6e · 7919c6e
1 parent 895bbec
commit 7919c6e
Showing 1 changed file with 1 addition and 5 deletions.
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
@@ -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
             )