fix: High/Medium audit bugs across state, environ, random, nn, transform - #244

Sorry @chaoming0625, you've used your own review budget of 250,000 diff characters for the last 7 days.

You can request another review in 6 days and 1 hour by commenting @sourcery-ai review. Upgrade to get a review now.