From 619a6df2ec78b085f1246472bb1c7729f72ae412 Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Thu, 13 Feb 2025 16:11:43 +0000 Subject: [PATCH] Update [ghstack-poisoned] --- tensordict/tensorclass.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/tensordict/tensorclass.py b/tensordict/tensorclass.py index 47de104ee..69dbfa4ae 100644 --- a/tensordict/tensorclass.py +++ b/tensordict/tensorclass.py @@ -1868,9 +1868,16 @@ def _repr(self) -> str: ",\n".join(non_tensor_fields), 4 * " ", ) - string = ",\n".join(field_str + [non_tensor_field_str, *medatada_fields]) - else: + if field_str: + string = ",\n".join(field_str + [non_tensor_field_str, *medatada_fields]) + else: + string = ",\n".join([non_tensor_field_str, *medatada_fields]) + elif field_str: string = ",\n".join(field_str + medatada_fields) + elif len(medatada_fields) > 0: + string = ",\n".join(medatada_fields) + else: + string = "" return f"{type(self).__name__}({string})"