fix(integrations): fix type check for jax.Array introduced in jax==0.4.1

[dmitryduev]

Description

See #4632 (comment) -- the sdk is currently not working with jax==0.4.1 bc of a jax API change.

Testing

How was this PR tested?

Checklist

• Include reference to internal ticket "Fixes WB-NNNN" and/or GitHub issue "Fixes #NNNN" (if applicable)
• Ensure PR title compliance with the conventional commits standards

[codecov]

Codecov Report

Merging #4718 (fc47b4b) into main (4f272d3) will increase coverage by 0.00%.
The diff coverage is 100.00%.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #4718   +/-   ##
=======================================
  Coverage   83.29%   83.30%           
=======================================
  Files         280      280           
  Lines       34946    34946           
=======================================
+ Hits        29108    29111    +3     
+ Misses       5838     5835    -3     

Flag	Coverage Δ
functest	55.65% <100.00%> (+0.02%)	⬆️
unittest	73.97% <100.00%> (+0.02%)	⬆️

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

Impacted Files	Coverage Δ
wandb/util.py	87.26% <100.00%> (ø)
wandb/sdk/lib/printer.py	88.60% <0.00%> (-0.64%)	⬇️
wandb/sdk/wandb_run.py	90.62% <0.00%> (-0.30%)	⬇️
wandb/sdk/lib/git.py	77.71% <0.00%> (ø)
wandb/integration/tensorboard/monkeypatch.py	91.35% <0.00%> (ø)
wandb/sdk/wandb_setup.py	88.94% <0.00%> (+0.50%)	⬆️
wandb/sdk/internal/system/system_info.py	91.55% <0.00%> (+3.24%)	⬆️
wandb/errors/term.py	93.47% <0.00%> (+6.52%)	⬆️

[raubitsj]

lgtm