Increase the number of shards for optimizer_pss_test, which is hitting timeout error on head.
Increase the number of shards for optimizer_pss_test, which is hitting timeout error on head.

PiperOrigin-RevId: 446848977

keras-team/keras,keras-team/keras