From 19d05c2961e437dcef67bfe4277115bf5cfb85a2 Mon Sep 17 00:00:00 2001 From: Mathijs de Boer Date: Tue, 21 Mar 2023 15:10:29 +0100 Subject: [PATCH 1/2] Prepend "meta" to `MetaTensor.__repr__` and `MetaTensor.__str__` for easier identification Signed-off-by: Mathijs de Boer --- monai/data/meta_tensor.py | 4 ++-- tests/test_meta_tensor.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/monai/data/meta_tensor.py b/monai/data/meta_tensor.py index 3bbd243b4a..8fb277df50 100644 --- a/monai/data/meta_tensor.py +++ b/monai/data/meta_tensor.py @@ -572,14 +572,14 @@ def __repr__(self): Prints a representation of the tensor identical to ``torch.Tensor.__repr__``. Use ``print_verbose`` for associated metadata. """ - return self.as_tensor().__repr__() + return f"meta{self.as_tensor().__repr__()}" def __str__(self): """ Prints a representation of the tensor identical to ``torch.Tensor.__str__``. Use ``print_verbose`` for associated metadata. """ - return str(self.as_tensor()) + return f"meta{str(self.as_tensor())}" def print_verbose(self) -> None: """Verbose print with meta data.""" diff --git a/tests/test_meta_tensor.py b/tests/test_meta_tensor.py index 4f2cb9636a..a6607a3ccd 100644 --- a/tests/test_meta_tensor.py +++ b/tests/test_meta_tensor.py @@ -430,8 +430,8 @@ def test_decollate(self, dtype): def test_str(self): t = MetaTensor([1.0], affine=torch.tensor(1), meta={"fname": "filename"}) - self.assertEqual(str(t), "tensor([1.])") - self.assertEqual(t.__repr__(), "tensor([1.])") + self.assertEqual(str(t), "metatensor([1.])") + self.assertEqual(t.__repr__(), "metatensor([1.])") def test_shape(self): s = MetaTensor([1]) From c2849d66622681802bdde1a111e5df75d94d6b6f Mon Sep 17 00:00:00 2001 From: Mathijs de Boer Date: Tue, 21 Mar 2023 15:18:02 +0100 Subject: [PATCH 2/2] Update docstrings to no longer say output is identical to torch.Tensor Signed-off-by: Mathijs de Boer --- monai/data/meta_tensor.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/monai/data/meta_tensor.py b/monai/data/meta_tensor.py index 8fb277df50..48b9320f99 100644 --- a/monai/data/meta_tensor.py +++ b/monai/data/meta_tensor.py @@ -569,14 +569,16 @@ def ensure_torch_and_prune_meta( def __repr__(self): """ - Prints a representation of the tensor identical to ``torch.Tensor.__repr__``. + Prints a representation of the tensor. + Prepends "meta" to ``torch.Tensor.__repr__``. Use ``print_verbose`` for associated metadata. """ return f"meta{self.as_tensor().__repr__()}" def __str__(self): """ - Prints a representation of the tensor identical to ``torch.Tensor.__str__``. + Prints a representation of the tensor. + Prepends "meta" to ``torch.Tensor.__str__``. Use ``print_verbose`` for associated metadata. """ return f"meta{str(self.as_tensor())}"