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

* [NFC] polish colossalai/booster/mixed_precision/mixed_precision_base.py code style
1 year ago · 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,