codestyle fix for hello-pt-resnet example.

examples/hello-world/hello-pt-resnet/fedavg_script_runner_pt.py

@@ -12,18 +12,21 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from src.simple_network import Resnet18
+
 from nvflare.app_opt.pt.job_config.fed_avg import FedAvgJob
 from nvflare.job_config.script_runner import ScriptRunner
-from src.simple_network import Resnet18
 
 if __name__ == "__main__":
     n_clients = 2
     num_rounds = 2
     train_script = "src/hello-pt_cifar10_fl.py"
 
     job = FedAvgJob(
-        name="hello-pt_cifar10_fedavg", n_clients=n_clients, num_rounds=num_rounds,
-        initial_model=Resnet18(num_classes=10)
+        name="hello-pt_cifar10_fedavg",
+        n_clients=n_clients,
+        num_rounds=num_rounds,
+        initial_model=Resnet18(num_classes=10),
     )
 
     # Add clients

examples/hello-world/hello-pt-resnet/src/hello-pt_cifar10_fl.py

@@ -15,6 +15,7 @@
 import os
 
 import torch
+from simple_network import Resnet18
 from torch import nn
 from torch.optim import SGD
 from torch.utils.data.dataloader import DataLoader
@@ -23,7 +24,6 @@
 
 import nvflare.client as flare
 from nvflare.client.tracking import SummaryWriter
-from simple_network import Resnet18
 
 DATASET_PATH = "/tmp/nvflare/data"
 

examples/hello-world/hello-pt-resnet/src/simple_network.py

@@ -11,15 +11,14 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from typing import Optional, Any
+from typing import Any, Optional
 
 from torchvision.models import ResNet
 from torchvision.models._utils import _ovewrite_named_param
 from torchvision.models.resnet import BasicBlock, ResNet18_Weights
 
 
 class Resnet18(ResNet):
-
     def __init__(self, num_classes, weights: Optional[ResNet18_Weights] = None, progress: bool = True, **kwargs: Any):
         self.num_classes = num_classes