[NFC] Fix format for mixed precision (#4253)

* [NFC] polish colossalai/booster/mixed_precision/mixed_precision_base.py code style
1 year ago · b366f1d99f
parent 02192a632e
commit b366f1d99f
1 changed files with 6 additions and 5 deletions
--- a/colossalai/booster/mixed_precision/mixed_precision_base.py
+++ b/colossalai/booster/mixed_precision/mixed_precision_base.py
@ -13,7 +13,8 @@ class MixedPrecision(ABC):
    """
    @abstractmethod
-    def configure(self,
+    def configure(
        self,
        model: nn.Module,
        optimizer: Optional[Optimizer] = None,
        criterion: Optional[Callable] = None,