Fix float errors

[eugeneteoh]

Description

This PR ...

Checklist

• Linked issues (if existing)
• Amended changelog for large changes (and added myself there as contributor)
• Added/modified tests
• Used pre-commit hooks when committing to ensure that code is compliant with hooks. Install hooks with pre-commit install.
  To run hooks independent of commit, execute pre-commit run --all-files

Make sure to have fun coding!

[jdb78]

Could you give an example? Would love to have a test to understand the problem better. Probably not ideal to just cast to float.