fix: doc summary for dict and set attributes (#1279)

Charlotte Gerhaher · web-flow · commit df6dc0bd0a81 · 2023-03-23T10:16:35.000+01:00
Signed-off-by: anna-charlotte &lt;charlotte.gerhaher@jina.ai&gt;
diff --git a/docarray/display/document_summary.py b/docarray/display/document_summary.py
@@ -137,14 +137,26 @@ def __rich_console__(
                 table.add_row(col_1, text.Text(col_2))
             elif isinstance(value, AbstractTensor):
                 table.add_row(col_1, TensorDisplay(tensor=value))
-            elif isinstance(value, (tuple, list)):
+            elif isinstance(value, (tuple, list, set)):
+                value_list = list(value)
                 col_2 = ''
-                for i, x in enumerate(value):
+                for i, x in enumerate(value_list):
                     if len(col_2) + len(str(x)) < 50:
-                        col_2 = str(value[:i])
+                        col_2 = str(value_list[: i + 1])
                     else:
-                        col_2 = f'{col_2[:-1]}, ...] (length: {len(value)})'
+                        col_2 = f'{col_2[:-1]}, ...] (length: {len(value_list)})'
                         break
+
+                if type(value) == tuple:
+                    col_2 = col_2.replace('[', '(', 1).replace(']', ')', -1)
+                if type(value) == set:
+                    col_2 = col_2.replace('[', '{', 1).replace(']', '}', -1)
+
+                table.add_row(col_1, text.Text(col_2))
+            elif isinstance(value, dict):
+                col_2 = f'{value}'
+                if len(col_2) > 50:
+                    col_2 = f'{col_2[: 50]}' + ' ... } ' + f'(length: {len(value)})'
                 table.add_row(col_1, text.Text(col_2))
 
         if table.rows:
