enh: add 'CG' as an optimizer

nipreps · Oct 26, 2024 · 10b1018 · 10b1018
1 parent 5fd0125
commit 10b1018
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/eddymotion/model/_sklearn.py b/src/eddymotion/model/_sklearn.py
@@ -109,7 +109,10 @@
 """Minimum nonzero angle."""
 LBFGS_CONFIGURABLE_OPTIONS = {"disp", "maxiter", "ftol", "gtol"}
 """The set of extended options that can be set on the default BFGS."""
-CONFIGURABLE_OPTIONS: Mapping[str, set] = {"Nelder-Mead": {"disp", "maxiter", "adaptive", "fatol"}}
+CONFIGURABLE_OPTIONS: Mapping[str, set] = {
+    "Nelder-Mead": {"disp", "maxiter", "adaptive", "fatol"},
+    "CG": {"disp", "maxiter", "gtol"},
+}
 """
 A mapping from optimizer names to the option set they allow.