update: test_bnb_optimizer

Author: kozistr

tests/test_create_optimizer.py

@@ -1,4 +1,6 @@
-from pytorch_optimizer import create_optimizer
+import pytest
+
+from pytorch_optimizer import create_optimizer, load_optimizer
 from tests.utils import LogisticRegression
 
 
@@ -7,3 +9,8 @@ def test_create_optimizer():
 
     create_optimizer(model, 'adamp', lr=1e-2, weight_decay=1e-3, use_gc=True, use_lookahead=True)
     create_optimizer(model, 'alig', lr=1e-2, use_lookahead=True)
+
+
+def test_bnb_optimizer():
+    with pytest.raises(NotImplementedError):
+        load_optimizer('bnb_adamw8bit')