Truncation during non-interactive model summary printing

[ryan-gustafson]

When printing a model summary non-interactively, the provided line_width is not honored, resulting in right-side truncation in the file context. The rich table's default of 80 seems to be used instead.

Changing:

keras/keras/src/utils/summary_utils.py

Line 353 in 6d26efb

highlight=False, force_terminal=False, color_system=None

Locally, if I set width to line_width, it produces the expected file contents:

   highlight=False, force_terminal=False, color_system=None, width=line_length

Example of non-interactive model printing with line_length > 80:

    with open(summary_path, 'w') as f:
        model.summary(expand_nested=True, show_trainable=True, line_length=120,
                      print_fn=lambda x: f.write(x + '\n'))

[dhantule]

Hi @ryan-gustafson, thanks for reporting this.

I have reproduced this issue and I also faced truncation when printing model summary non interactively with line_length>80

Model: "sequential_24"
┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━┳━━━━━━━
┃ Layer (type)                                  ┃ Output Shape                        ┃            Param # ┃ Traina
┡━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━╇━━━━━━━
│ dense_48 (Dense)                              │ (None, 28, 128)                     │              3,712 │     Y 
├───────────────────────────────────────────────┼─────────────────────────────────────┼────────────────────┼───────
│ dense_49 (Dense)                              │ (None, 28, 10)                      │              1,290 │     Y 
└───────────────────────────────────────────────┴─────────────────────────────────────┴────────────────────┴───────
 Total params: 5,002 (19.54 KB)
 Trainable params: 5,002 (19.54 KB)
 Non-trainable params: 0 (0.00 B)

Can you open a PR?

[github-actions]

This issue is stale because it has been open for 14 days with no activity. It will be closed if no further activity occurs. Thank you.