Fix edge cases in history (jax scalar) and add tests (!619) · Merge requests · Vicentini Filippo / netket

Vicentini Filippo requested to merge impro-history into master Apr 12, 2021

Created by: PhilipVinc

Previously History was not handling properly jax scalars. Now it does, and a bunch of tests are added

Fix edge cases in history (jax scalar) and add tests