Add for 'iota'

alphacsc · May 16, 2023 · e044cfd · e044cfd
1 parent 2d8c951
commit e044cfd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/alphacsc/_z_encoder.py b/alphacsc/_z_encoder.py
@@ -36,7 +36,7 @@ def get_z_encoder_for(X, D_hat, n_atoms, n_times_atom, n_jobs,
         The number of parallel jobs.
     solver : str
         The solver to use for the z update. Options are
-        {{'l_bfgs' (default) | 'lgcd' | 'fista' | 'dicodile'}}.
+        {{'l_bfgs' (default) | 'lgcd' | 'fista' | 'ista' | 'dicodile'}}.
     solver_kwargs : dict
         Additional keyword arguments to pass to update_z_multi.
     reg : float
@@ -63,7 +63,7 @@ def get_z_encoder_for(X, D_hat, n_atoms, n_times_atom, n_jobs,
 
     assert reg is not None, 'reg value cannot be None.'
 
-    if solver in ['l-bfgs', 'lgcd', 'fista']:
+    if solver in ['l-bfgs', 'lgcd', 'fista', 'ista']:
 
         return AlphaCSCEncoder(
             X, D_hat, n_atoms, n_times_atom, n_jobs,

diff --git a/alphacsc/convolutional_dictionary_learning.py b/alphacsc/convolutional_dictionary_learning.py
@@ -69,7 +69,7 @@
 
     solver_z : str
         The solver to use for the z update. Options are
-        {{'l_bfgs' (default) | 'lgcd', 'fista'}}.
+        {{'l_bfgs' (default) | 'lgcd', 'fista', 'ista'}}.
     solver_z_kwargs : dict
         Additional keyword arguments to pass to update_z_multi.
     unbiased_z_hat : boolean