From c76a77e9765f18b2518b3662c6e7ae6d16a1ab7a Mon Sep 17 00:00:00 2001 From: vfdev-5 Date: Wed, 12 Oct 2022 15:16:54 +0000 Subject: [PATCH] Fixed repr for ElasticTransform --- torchvision/transforms/transforms.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/torchvision/transforms/transforms.py b/torchvision/transforms/transforms.py index 18e2ffc9680..6011e237279 100644 --- a/torchvision/transforms/transforms.py +++ b/torchvision/transforms/transforms.py @@ -2133,9 +2133,9 @@ def forward(self, tensor: Tensor) -> Tensor: return F.elastic_transform(tensor, displacement, self.interpolation, self.fill) def __repr__(self): - format_string = self.__class__.__name__ + "(alpha=" - format_string += str(self.alpha) + ")" - format_string += ", (sigma=" + str(self.sigma) + ")" - format_string += ", interpolation={self.interpolation}" - format_string += ", fill={self.fill})" + format_string = self.__class__.__name__ + format_string += f"(alpha={self.alpha}" + format_string += f", sigma={self.sigma}" + format_string += f", interpolation={self.interpolation}" + format_string += f", fill={self.fill})" return format_string