Arm backend: Format docs in backends/arm/test/models

Signed-off-by: Sebastian Larsson <sebastian.larsson@arm.com>
Change-Id: I655f7e88eccd3226bc27aae4a2dc1804a37f6506

Author: Sebastian-Larsson

backends/arm/test/models/stable_diffusion/test_CLIPTextModelWithProjection.py

@@ -30,9 +30,11 @@
 
 
 class TestCLIPTextModelWithProjection:
-    """
-    Test class of CLIPTextModelWithProjection.
-    CLIPTextModelWithProjection is one of the text_encoder used by Stable Diffusion 3.5 Medium
+    """Test class of CLIPTextModelWithProjection.
+
+    CLIPTextModelWithProjection is one of the text_encoder used by Stable
+    Diffusion 3.5 Medium
+
     """
 
     # Adjust nbr below as we increase op support.

backends/arm/test/models/stable_diffusion/test_SD3Transformer2DModel.py

@@ -25,9 +25,11 @@
 
 
 class TestSD3Transformer2DModel:
-    """
-    Test class of AutoenSD3Transformer2DModelcoderKL.
-    SD3Transformer2DModel is the transformer model used by Stable Diffusion 3.5 Medium
+    """Test class of AutoenSD3Transformer2DModelcoderKL.
+
+    SD3Transformer2DModel is the transformer model used by Stable Diffusion 3.5
+    Medium
+
     """
 
     # Adjust nbr below as we increase op support.

backends/arm/test/models/stable_diffusion/test_T5EncoderModel.py

@@ -28,9 +28,11 @@
 
 
 class TestT5EncoderModel:
-    """
-    Test class of T5EncoderModel.
-    T5EncoderModel is one of the text_encoder used by Stable Diffusion 3.5 Medium
+    """Test class of T5EncoderModel.
+
+    T5EncoderModel is one of the text_encoder used by Stable Diffusion 3.5
+    Medium
+
     """
 
     # Adjust nbr below as we increase op support.

backends/arm/test/models/stable_diffusion/test_vae_AutoencoderKL.py

@@ -28,9 +28,10 @@
 
 
 class TestAutoencoderKL:
-    """
-    Test class of AutoencoderKL.
+    """Test class of AutoencoderKL.
+
     AutoencoderKL is the encoder/decoder used by Stable Diffusion 3.5 Medium
+
     """
 
     def _prepare_inputs(self, batch_size=4, num_channels=3, sizes=(32, 32)):

backends/arm/test/models/test_conformer.py

@@ -28,7 +28,7 @@ def get_test_inputs(dim, lengths, num_examples):
 
 
 class TestConformer:
-    """Tests Torchaudio Conformer"""
+    """Tests Torchaudio Conformer."""
 
     # Adjust nbr below as we increase op support. Note: most of the delegates
     # calls are directly consecutive to each other in the .pte. The reason

backends/arm/test/models/test_llama.py

@@ -42,11 +42,13 @@
 
 
 class TestLlama:
-    """
-    Test class of Llama models. Type of Llama model depends on command line parameters:
+    """Test class of Llama models.
+
+    Type of Llama model depends on command line parameters:
     --llama_inputs <path to .pt file> <path to json file> <name of model variant>
     Example: --llama_inputs stories110M/stories110M.pt stories110M/params.json stories110m
     For more examples and info see examples/models/llama/README.md.
+
     """
 
     def prepare_model(self):
@@ -101,7 +103,7 @@ def prepare_model(self):
 
 
 def _use_partial_quantizer(pipeline):
-    """Set the pipeline's quantizer to only include Linear layers"""
+    """Set the pipeline's quantizer to only include Linear layers."""
     pipeline.quantizer.set_global(None)
     pipeline.quantizer.set_module_type(
         torch.nn.Linear, get_symmetric_quantization_config()

backends/arm/test/models/test_lstm_arm.py

@@ -143,7 +143,7 @@ def test_lstm_vgf_no_quant():
 
 
 def test_lstm_tosa_INT_16a8w():
-    """Test LSTM model with 16A8W quantization (16-bit activations, 8-bit weights)"""
+    """Test LSTM 16A8W quantization (16-bit activations, 8-bit weights)."""
 
     pipeline = TosaPipelineINT[input_t](
         TestLSTM.lstm,

backends/arm/test/models/test_mobilenet_v2_arm.py

@@ -41,7 +41,7 @@
 
 
 def _use_partial_quantizer(pipeline):
-    """Set the pipeline's quantizer to only include Conv2d and ReLU6"""
+    """Set the pipeline's quantizer to only include Conv2d and ReLU6."""
     quant_cfg = get_symmetric_quantization_config()
     pipeline.quantizer.set_global(None)
     pipeline.quantizer.set_module_type(torch.nn.Conv2d, quant_cfg)

backends/arm/test/models/test_nn_functional.py

@@ -3,8 +3,9 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-"""
-Tests 10 popular torch.nn.functional not tested in other ways or training related
+"""Tests 10 popular torch.nn.functional not tested in other ways or training
+related.
+
 - normalize
 - grid_sample
 - one_hot
@@ -16,6 +17,7 @@
 - affine_grid
 - max_pool1d
 - threshold
+
 """
 from typing import Callable
 

backends/arm/test/models/test_nn_modules.py

@@ -3,8 +3,8 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-"""
-Tests 10 popular nn modules not tested in other ways or training related.
+"""Tests 10 popular nn modules not tested in other ways or training-related.
+
 - Embedding
 - LeakyReLU
 - BatchNorm1d
@@ -15,6 +15,7 @@
 - InstanceNorm2d
 - PReLU
 - Transformer
+
 """
 
 from typing import Callable