fix more linting errors

src/itwinai/torch/monitoring/plotting.py

@@ -1,4 +1,4 @@
-from typing import Tuple, List
+from typing import Tuple
 
 import matplotlib
 import matplotlib.pyplot as plt

tutorials/distributed-ml/torch-scaling-test/ddp_trainer.py

@@ -1,28 +1,25 @@
 """
 Scaling test of torch Distributed Data Parallel on Imagenet using Resnet.
 """
-from typing import Optional
 import argparse
-import sys
 import os
-from timeit import default_timer as timer
+import sys
 import time
+from timeit import default_timer as timer
+from typing import Optional
 
 import torch
 import torch.distributed as dist
 import torch.nn as nn
 import torch.nn.functional as F
+import torchvision
 from torch.utils.data import DataLoader
 from torch.utils.data.distributed import DistributedSampler
-import torchvision
+from utils import imagenet_dataset
 
-from itwinai.parser import ArgumentParser as ItAIArgumentParser
 from itwinai.loggers import EpochTimeTracker
-from itwinai.torch.reproducibility import (
-    seed_worker, set_seed
-)
-
-from utils import imagenet_dataset
+from itwinai.parser import ArgumentParser as ItAIArgumentParser
+from itwinai.torch.reproducibility import seed_worker, set_seed
 
 
 def parse_params():

tutorials/distributed-ml/torch-scaling-test/deepspeed_trainer.py

@@ -1,28 +1,25 @@
 """
 Scaling test of Microsoft Deepspeed on Imagenet using Resnet.
 """
-from typing import Optional
 import argparse
-import sys
 import os
-from timeit import default_timer as timer
+import sys
 import time
-import deepspeed
+from timeit import default_timer as timer
+from typing import Optional
 
+import deepspeed
 import torch
 import torch.distributed as dist
 import torch.nn.functional as F
+import torchvision
 from torch.utils.data import DataLoader
 from torch.utils.data.distributed import DistributedSampler
-import torchvision
+from utils import imagenet_dataset
 
-from itwinai.parser import ArgumentParser as ItAIArgumentParser
 from itwinai.loggers import EpochTimeTracker
-from itwinai.torch.reproducibility import (
-    seed_worker, set_seed
-)
-
-from utils import imagenet_dataset
+from itwinai.parser import ArgumentParser as ItAIArgumentParser
+from itwinai.torch.reproducibility import seed_worker, set_seed
 
 
 def parse_params():

tutorials/distributed-ml/torch-scaling-test/horovod_trainer.py

@@ -1,29 +1,27 @@
 """
 Scaling test of Horovod on Imagenet using Resnet.
 """
-from typing import Optional
 import argparse
 import os
 import sys
-from timeit import default_timer as timer
 import time
+from timeit import default_timer as timer
+from typing import Optional
 
+import horovod.torch as hvd
 import torch
+
 # import torch.multiprocessing as mp
 import torch.nn.functional as F
 import torch.optim as optim
+import torchvision
 from torch.utils.data import DataLoader
 from torch.utils.data.distributed import DistributedSampler
-import horovod.torch as hvd
-import torchvision
+from utils import imagenet_dataset
 
-from itwinai.parser import ArgumentParser as ItAIArgumentParser
 from itwinai.loggers import EpochTimeTracker
-from itwinai.torch.reproducibility import (
-    seed_worker, set_seed
-)
-
-from utils import imagenet_dataset
+from itwinai.parser import ArgumentParser as ItAIArgumentParser
+from itwinai.torch.reproducibility import seed_worker, set_seed
 
 
 def parse_params():