diff --git a/monai/data/meta_tensor.py b/monai/data/meta_tensor.py index 3bbd243b4a..48b9320f99 100644 --- a/monai/data/meta_tensor.py +++ b/monai/data/meta_tensor.py @@ -569,17 +569,19 @@ 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 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__``. + Prints a representation of the tensor. + Prepends "meta" 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])