filter step size warning

geoopt · Feb 27, 2021 · 89d5a0f · 89d5a0f
1 parent 3730c3c
commit 89d5a0f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_rlinesearch.py b/tests/test_rlinesearch.py
@@ -10,6 +10,7 @@
 @pytest.mark.parametrize("batch_size", [None, 1, 16])
 @pytest.mark.parametrize("line_search_method", ["armijo", "wolfe"])
 @pytest.mark.parametrize("cg_method", ["steepest", "fr", "pr"])
+@pytest.mark.filterwarnings("ignore:No suitable step size could be found")
 def test_rwolfe_stiefel(line_search_params, batch_size, line_search_method, cg_method):
     # Use line search to solve orthogonal procrustes
     stiefel = geoopt.manifolds.Stiefel()