🎨 Fix format

Author: ZihaoZhou

.github/workflows/flake8-link.yml

@@ -8,11 +8,11 @@ stages:
 build-and-push-job:
   stage: build-and-push
   script:
-    - echo "{\"auths\":{\"$CI_REGISTRY\":{\"username\":\"$CI_REGISTRY_USER\",\"password\":\"$CI_REGISTRY_PASSWORD\"}}}" 
+    - echo "{\"auths\":{\"$CI_REGISTRY\":{\"username\":\"$CI_REGISTRY_USER\",\"password\":\"$CI_REGISTRY_PASSWORD\"}}}"
       > /kaniko/.docker/config.json
     - /kaniko/executor --cache=true
-      --context $CI_PROJECT_DIR 
-      --dockerfile $CI_PROJECT_DIR/Dockerfile 
-      --destination $CI_REGISTRY_IMAGE:${CI_COMMIT_SHA:0:8} 
+      --context $CI_PROJECT_DIR
+      --dockerfile $CI_PROJECT_DIR/Dockerfile
+      --destination $CI_REGISTRY_IMAGE:${CI_COMMIT_SHA:0:8}
       --destination $CI_REGISTRY_IMAGE:temp
   timeout: 10h

.gitlab-ci.yml

@@ -5,11 +5,6 @@ ci:
   autoupdate_schedule: monthly
 
 repos:
-  - repo: https://github.com/psf/black
-    rev: 24.4.0
-    hooks:
-      - id: black
-
   - repo: https://github.com/asottile/pyupgrade
     rev: v3.15.2
     hooks:
@@ -22,6 +17,14 @@ repos:
       - id: isort
         additional_dependencies: [toml]
 
+  - repo: https://github.com/pycqa/flake8
+    rev: 7.0.0
+    hooks:
+      - id: flake8
+        args:
+          - --max-line-length=150
+          - --ignore=E402,E731,F541,W291,E122,E127,F401,E266,E241,C901,E741,W293,F811,W503,E203,F403,F405
+
   - repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v4.6.0
     hooks:

.pre-commit-config.yaml

@@ -2,16 +2,17 @@
 # Reference: https://blog.csdn.net/u010900574/article/details/122780585 #
 #########################################################################
 
-from __future__ import print_function
+import argparse
 import os
 import time
-import argparse
+
 import torch
+import torch.distributed as dist
 import torch.nn.functional as F
 import torch.optim as optim
-from torchvision import datasets, transforms
 from torch.optim.lr_scheduler import StepLR
-import torch.distributed as dist
+from torchvision import datasets, transforms
+
 from net import Net
 
 

mnist_ddp.py

@@ -1,13 +1,13 @@
-from __future__ import print_function
-import time
 import argparse
+import time
+
+import deepspeed
 import torch
 import torch.nn.functional as F
-from torchvision import datasets, transforms
-from net import Net
-import deepspeed
 from deepspeed import comm as dist
+from torchvision import datasets, transforms
 
+from net import Net
 
 deepspeed.init_distributed()
 

mnist_ds.py

@@ -1,15 +1,16 @@
-from __future__ import print_function
+import argparse
 import os
 import time
-import argparse
+
 import torch
 import torch.nn.functional as F
 import torch.optim as optim
-from torchvision import datasets, transforms
-from torch.optim.lr_scheduler import StepLR
-from net import Net
 from accelerate import Accelerator
 from accelerate.state import AcceleratorState
+from torch.optim.lr_scheduler import StepLR
+from torchvision import datasets, transforms
+
+from net import Net
 
 accelerator = Accelerator()
 local_rank = int(os.environ.get("LOCAL_RANK", -1))

mnist_hf.py

@@ -4,13 +4,15 @@
 
 import argparse
 import time
+
+import horovod.torch as hvd
 import torch
 import torch.multiprocessing as mp
 import torch.nn.functional as F
 import torch.optim as optim
-from torchvision import datasets, transforms
 import torch.utils.data.distributed as dist
-import horovod.torch as hvd
+from torchvision import datasets, transforms
+
 from net import Net
 
 # Training settings

mnist_hvd.py

@@ -5,7 +5,7 @@
 
 class Net(nn.Module):
     def __init__(self):
-        super(Net, self).__init__()
+        super().__init__()
         self.conv1 = nn.Conv2d(1, 32, 3, 1)
         self.conv2 = nn.Conv2d(32, 64, 3, 1)
         self.dropout1 = nn.Dropout(0.25)