fix totensorv2 docstring (#1400)

Co-authored-by: Mikhail Druzhinin <dipetm@gmail.com>
albumentations-team · Mar 8, 2023 · 9b0525f · 9b0525f
1 parent 227bd74
commit 9b0525f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/albumentations/pytorch/transforms.py b/albumentations/pytorch/transforms.py
@@ -65,9 +65,11 @@ class ToTensorV2(BasicTransform):
     instead).
 
     Args:
-        transpose_mask (bool): if True and an input mask has three dimensions, this transform will transpose dimensions
-        so the shape `[height, width, num_channels]` becomes `[num_channels, height, width]`. The latter format is a
-        standard format for PyTorch Tensors. Default: False.
+        transpose_mask (bool): If True and an input mask has three dimensions, this transform will transpose dimensions
+            so the shape `[height, width, num_channels]` becomes `[num_channels, height, width]`. The latter format is a
+            standard format for PyTorch Tensors. Default: False.
+        always_apply (bool): Indicates whether this transformation should be always applied. Default: True.
+        p (float): Probability of applying the transform. Default: 1.0.
     """
 
     def __init__(self, transpose_mask=False, always_apply=True, p=1.0):