fix: typo

kozistr · May 5, 2024 · 5a5cc99 · 5a5cc99
1 parent 546011e
commit 5a5cc99
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_optimizers.py b/tests/test_optimizers.py
@@ -467,7 +467,7 @@ def test_reset(optimizer_config):
     optimizer_class, config, _ = optimizer_config
     if optimizer_class.__name__ == 'Ranger21':
         config.update({'num_iterations': 1})
-    elif optimizer_class.__name__ == 'bSAM':
+    elif optimizer_class.__name__ == 'BSAM':
         config.update({'num_data': 1})
 
     optimizer = optimizer_class([simple_parameter()], **config)