update: test_scion_lmo_types

Author: kozistr

tests/test_optimizers.py

@@ -1007,6 +1007,10 @@ def test_scion_lmo_types():
         norm.init(grad)
         norm.lmo(grad)
 
+    norm = build_lmo_norm(norm_type=2, max_scale=True)
+    norm.init(grad_2d)
+    norm.lmo(grad_2d)
+
     with pytest.raises(NotImplementedError):
         norm.init(grad_5d)