Fix match case.

mila-iqia · Dec 3, 2024 · 895bbec · 895bbec
1 parent 2f6bbb3
commit 895bbec
Showing 1 changed file with 6 additions and 1 deletion.
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
@@ -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":