Merge pull request #110 from mila-iqia/hotfix

Fix match case.
mila-iqia · Dec 3, 2024 · 505a833 · 505a833
2 parents 2f6bbb3 + 7919c6e
commit 505a833
Showing 1 changed file with 3 additions and 2 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
@@ -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
             )