fix lint

examples/mnist_ddp.py

@@ -1,7 +1,10 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
-"""The ddp mnist example using MS-AMP. It is adapted from https://github.com/pytorch/examples/blob/main/mnist/main.py."""
+"""The ddp mnist example using MS-AMP.
+
+It is adapted from https://github.com/pytorch/examples/blob/main/mnist/main.py.
+"""
 
 from __future__ import print_function
 import os

examples/mnist_fsdp.py

@@ -73,9 +73,9 @@ def train(args, model, rank, world_size, train_loader, optimizer, epoch, sampler
         sampler.set_epoch(epoch)
     for batch_idx, (data, target) in enumerate(train_loader):
         data, target = data.to(rank), target.to(rank)
-        
+
         optimizer.zero_grad()
-        
+
         output = model(data)
         loss = F.nll_loss(output, target, reduction='sum')
         loss.backward()