From 2b40fbd6dab1aa20e930e1e08a91b0246f7d9c4d Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Fri, 9 Feb 2024 15:31:01 +0100
Subject: [PATCH] [models.transforms] Fix qa
---
src/mednet/models/transforms.py | 8 +++-----
1 file changed, 3 insertions(+), 5 deletions(-)
diff --git a/src/mednet/models/transforms.py b/src/mednet/models/transforms.py
index afb158ac..2a4f933a 100644
--- a/src/mednet/models/transforms.py
+++ b/src/mednet/models/transforms.py
@@ -10,7 +10,7 @@ import torchvision.transforms.functional
def square_center_pad(img: torch.Tensor) -> torch.Tensor:
- """Returns a squared version of the image, centered on a canvas padded with
+ """Return a squared version of the image, centered on a canvas padded with
zeros.
Parameters
@@ -23,8 +23,7 @@ def square_center_pad(img: torch.Tensor) -> torch.Tensor:
Returns
-------
- img
- transformed tensor, guaranteed to be square (ie. equal height and
+ Transformed tensor, guaranteed to be square (ie. equal height and
width).
"""
@@ -129,8 +128,7 @@ def rgb_to_grayscale(img: torch.Tensor) -> torch.Tensor:
class SquareCenterPad(torch.nn.Module):
- """Transforms to a squared version of the image, centered on a canvas
- padded with zeros."""
+ """Transform to a squared version of the image, centered on a canvas padded with zeros."""
def __init__(self):
super().__init__()
--
GitLab