fix imports and number of args violations

!62667 (merged) inspired to me run python checker on our code, some more fixes to the non-existing imports and mismatches in the fcn calls.

It is not a complete list of fixes - ignoring legacy config and focusing on tracking