Transition from DeviceArray to jax.Array

[bwohlberg]

Transition from DeviceArray to jax.Array, the new JAX array type from 0.4.1 onwards.

[codecov]

Codecov Report

Merging #410 (902efe7) into main (418865f) will increase coverage by 0.28%.
The diff coverage is 99.50%.

@@            Coverage Diff             @@
##             main     #410      +/-   ##
==========================================
+ Coverage   94.43%   94.71%   +0.28%     
==========================================
  Files          83       83              
  Lines        5103     5093      -10     
==========================================
+ Hits         4819     4824       +5     
+ Misses        284      269      -15     

Flag	Coverage Δ
unittests	94.71% <99.50%> (+0.28%)	⬆️

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

Impacted Files	Coverage Δ
scico/typing.py	100.00% <ø> (ø)
scico/linop/_convolve.py	86.31% <83.33%> (ø)
scico/__init__.py	100.00% <100.00%> (ø)
scico/data/__init__.py	100.00% <100.00%> (ø)
scico/denoiser.py	88.67% <100.00%> (-0.11%)	⬇️
scico/examples.py	96.00% <100.00%> (ø)
scico/flax/_flax.py	100.00% <100.00%> (ø)
scico/flax/blocks.py	100.00% <100.00%> (ø)
scico/flax/examples/data_generation.py	74.82% <100.00%> (-0.72%)	⬇️
scico/flax/examples/data_preprocessing.py	94.96% <100.00%> (+0.03%)	⬆️
... and 49 more

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.

[Michael-T-McCann]

Pending results of off-github discussion on snp.Array vs scico.typing.Array

[bwohlberg]

See also #181.