Just have your harness rules add types and run ty after every change. Models handle Python typing quite well these days.