flax

Fix failing CI jobs: trailing whitespace, deprecated `.type` usage

#4823

Merged