update: test_closure

kozistr · kozistr · commit becbd9183829 · 2023-01-01T20:46:54.000+09:00
diff --git a/tests/test_optimizers.py b/tests/test_optimizers.py
@@ -358,7 +358,7 @@ def test_closure(optimizer):
 
     try:
         optimizer.step(closure=dummy_closure)
-    except ZeroDivisionError:  # in case of Ranger21, Adai optimizers
+    except ValueError:  # in case of Ranger21, Adai optimizers
         pass
 
 
