Precision check feature

[trsav]

Type of changes

• Bug fix
• New feature
• Documentation / docstrings
• Tests
• Other

Checklist

• I've formatted the new code by running poetry run pre-commit run --all-files --show-diff-on-failure before committing.
• I've added tests for new code.
• I've added docstrings for the new code.

Description

Have added a warning using the warnings package that informs when X or y isn't of dtype float64 (if the warnings package isn't appropriate then please let me know, it's a new import within dataset.py). This is called immediately after checking the dataset shape:

    def __post_init__(self) -> None:
        r"""Checks that the shapes of $`X`$ and $`y`$ are compatible,
        and provides warnings regarding the precision of $`X`$ and $`y`$."""
        _check_shape(self.X, self.y)
       -->>  _check_precision(self.X, self.y)

I've added a test for the warning generation, but to do so I had to enable the higher precision within test_dataset.py (config.update("jax_enable_x64", True)) . I hope this solution is alright. Thanks!

Issue Number: #342

[thomaspinder]

Wonderful. Thanks for such a clean PR @trsav!