refactor: NoClosureError

Author: kozistr

pytorch_optimizer/base/exception.py

@@ -19,7 +19,7 @@ def __init__(self):
         super().__init__(self.message)
 
 
-class ClosureError(Exception):
+class NoClosureError(Exception):
     """Raised when there's no closure function"""
 
     def __init__(self, optimizer_name: str):

pytorch_optimizer/optimizer/sam.py

@@ -4,7 +4,7 @@
 from torch.optim.optimizer import Optimizer
 
 from pytorch_optimizer.base.base_optimizer import BaseOptimizer
-from pytorch_optimizer.base.exception import ClosureError
+from pytorch_optimizer.base.exception import NoClosureError
 from pytorch_optimizer.base.types import CLOSURE, DEFAULTS, OPTIMIZER, PARAMETERS
 
 
@@ -124,7 +124,7 @@ def second_step(self, zero_grad: bool = False):
     @torch.no_grad()
     def step(self, closure: CLOSURE = None):
         if closure is None:
-            raise ClosureError(self.__name__)
+            raise NoClosureError(self.__name__)
 
         self.first_step(zero_grad=True)