fix format (#586)

colossalai/engine/gradient_handler/_base_gradient_handler.py

@@ -12,6 +12,7 @@ class BaseGradientHandler(ABC):
         model (Module): Model where the gradients accumulate.
         optimizer (Optimizer): Optimizer for updating the parameters.
     """
+
     def __init__(self, model, optimizer):
         self._model = model
         self._optimizer = optimizer

colossalai/engine/ophooks/_base_ophook.py

@@ -5,6 +5,7 @@ import torch
 class BaseOpHook(ABC):
     """This class allows users to add customized operations
     before and after the execution of a PyTorch submodule"""
+
     def __init__(self):
         pass