diff --git a/tutorials/distributed-ml/torch-scaling-test/itwinai_trainer.py b/tutorials/distributed-ml/torch-scaling-test/itwinai_trainer.py
index 69d3533b..b1ee4ea9 100644
--- a/tutorials/distributed-ml/torch-scaling-test/itwinai_trainer.py
+++ b/tutorials/distributed-ml/torch-scaling-test/itwinai_trainer.py
@@ -20,14 +20,12 @@
 
 import horovod.torch as hvd
 import torch
-
 import torchvision
 from torch.utils.data import DataLoader
 from torch.utils.data.distributed import DistributedSampler
 from utils import get_parser, imagenet_dataset, train_epoch
 
 from itwinai.loggers import EpochTimeTracker
-
 from itwinai.torch.distributed import (
     DeepSpeedStrategy,
     HorovodStrategy,