Update dtype test for jax

[grasskin]

No description provided.

[codecov-commenter]

Codecov Report

Attention: 2 lines in your changes are missing coverage. Please review.

Comparison is base (c8700f4) 80.14% compared to head (c2a960a) 75.52%.
Report is 9 commits behind head on master.

Files	Patch %	Lines
keras/backend/jax/trainer.py	94.73%	0 Missing and 2 partials ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##           master   #19141      +/-   ##
==========================================
- Coverage   80.14%   75.52%   -4.62%     
==========================================
  Files         341      357      +16     
  Lines       36163    38447    +2284     
  Branches     7116     7483     +367     
==========================================
+ Hits        28982    29037      +55     
- Misses       5578     7806    +2228     
- Partials     1603     1604       +1     

Flag	Coverage Δ
keras	75.38% <98.27%> (-4.61%)	⬇️
keras-jax	59.38% <65.51%> (-3.68%)	⬇️
keras-numpy	53.75% <29.31%> (-3.34%)	⬇️
keras-tensorflow	60.87% <38.79%> (-3.79%)	⬇️
keras-torch	60.12% <29.31%> (-3.75%)	⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.