Sharded weights type error#2296

Merged

sachinprasadhs merged 4 commits intokeras-team:masterfrom

laxmareddyp:flatten_weight_map_values

Jun 16, 2025

Commits on Jun 11, 2025

fix-sharded-weights-typeerror
laxmareddyp
committed

Commits on Jun 12, 2025

Commits on Jun 13, 2025

Moved common initialization code to setUp for cleaner and Updated all three relevant test cases to use the shared setup.
laxmareddyp
committed