Several clean-ups

See merge request !25

Using style from https://github.com/ray-project/ray/pull/28376

The extension code would need to be added for this to work.