fix: support data generation with jax

[redeboer]

Simple fix: just casts ScalarSequence back into a numpy array so that a lambdified Function can handle it.

[redeboer]

Next step would be to allow feeding an Estimator a LambdifiedFunction directly. This is useful if lambdifying takes a long time.

[codecov]

Codecov Report

Merging #271 (943bb4b) into main (66e24e0) will increase coverage by 0.04%.
The diff coverage is 100.00%.

@@            Coverage Diff             @@
##             main     #271      +/-   ##
==========================================
+ Coverage   84.08%   84.12%   +0.04%     
==========================================
  Files          13       13              
  Lines         754      756       +2     
==========================================
+ Hits          634      636       +2     
  Misses        120      120              

Flag	Coverage Δ
unittests	84.12% <100.00%> (+0.04%)	⬆️

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

Impacted Files	Coverage Δ
src/tensorwaves/data/transform.py	100.00% <100.00%> (ø)