Update to make pydocstyle --numpy compliant.

Cambridge-ICCS · Nov 10, 2023 · f61cdcc · f61cdcc
1 parent 9ead87d
commit f61cdcc
Show file tree

Hide file tree

Showing 6 changed files with 2 additions and 9 deletions.
diff --git a/examples/1_SimpleNet/simplenet.py b/examples/1_SimpleNet/simplenet.py
@@ -1,4 +1,4 @@
-"""Module defining a simple PyTorch 'Net' for coupling to Fortran. """
+"""Module defining a simple PyTorch 'Net' for coupling to Fortran."""
 import torch
 from torch import nn
 
@@ -15,7 +15,6 @@ def __init__(
         Consists of a single Linear layer with weights predefined to
         multiply the input by 2.
         """
-
         super().__init__()
         self._fwd_seq = nn.Sequential(
             nn.Linear(5, 5, bias=False),
@@ -38,7 +37,6 @@ def forward(self, batch: torch.Tensor) -> torch.Tensor:
             batch scaled by 2.
 
         """
-
         return self._fwd_seq(batch)
 
 

diff --git a/examples/1_SimpleNet/simplenet_infer_fortran b/examples/1_SimpleNet/simplenet_infer_fortran
diff --git a/examples/1_SimpleNet/simplenet_infer_python.py b/examples/1_SimpleNet/simplenet_infer_python.py
@@ -21,7 +21,6 @@ def deploy(saved_model: str, device: str, batch_size: int = 1) -> torch.Tensor:
     output : torch.Tensor
         result of running inference on model with example Tensor input
     """
-
     input_tensor = torch.tensor([0.0, 1.0, 2.0, 3.0, 4.0]).repeat(batch_size, 1)
 
     if device == "cpu":

diff --git a/examples/2_ResNet18/resnet18.py b/examples/2_ResNet18/resnet18.py
@@ -21,7 +21,6 @@ def initialize(precision: torch.dtype) -> torch.nn.Module:
     model: torch.nn.Module
         Pretrained ResNet-18 model
     """
-
     # Set working precision
     torch.set_default_dtype(precision)
 
@@ -96,7 +95,7 @@ def run_model(model: torch.nn.Module, precision: type) -> None:
 
 
 def print_top_results(output: torch.Tensor) -> None:
-    """Prints top 5 results
+    """Print top 5 results.
 
     Parameters
     ----------

diff --git a/examples/n_c_and_cpp/resnet18.py b/examples/n_c_and_cpp/resnet18.py
@@ -14,7 +14,6 @@ def initialize():
     -------
     model : torch.nn.Module
     """
-
     # Load a pre-trained PyTorch model
     print("Loading pre-trained ResNet-18 model...", end="")
     model = torchvision.models.resnet18(pretrained=True)
@@ -35,7 +34,6 @@ def run_model(model):
     ----------
     model : torch.nn.Module
     """
-
     print("Running ResNet-18 model for ones...", end="")
     dummy_input = torch.ones(1, 3, 224, 224)
     output = model(dummy_input)

diff --git a/examples/n_c_and_cpp/resnet_infer_python.py b/examples/n_c_and_cpp/resnet_infer_python.py
@@ -22,7 +22,6 @@ def deploy(saved_model, device, batch_size=1):
     output : torch.Tensor
         result of running inference on model with Tensor of ones
     """
-
     image_filename = "data/dog.jpg"
     input_image = Image.open(image_filename)
     preprocess = torchvision.transforms.Compose(