Incorrect Symbolic Output Shape for Discretization Layer with output_mode="count"/"multi_hot"

[Isa-Fay]

When using Keras' Discretization layer with output_mode set to either "count" or "multi_hot", the symbolic output shape remains (None, 4) regardless of the number of bin boundaries configured.
Here's an example in this gist

[dataCenter430]

Hi, @sachinprasadhs
I'm a big fan of Keras and I'd love to have the chance to contribute here/
May I pick this up to reduce your workload?
Thanks

[dhantule]

Hi @Isa-Fay,

I have reproduced this issue with Keras 3.113.2 in this gist.

@dataCenter430, feel free to raise a PR with a fix.

[dataCenter430]

okay, I'm on it, thanks