move classes to module to use __target__ feature (#14023) (#14031)

* move classes to module to use __target__ feature

* Apply isort and black reformatting

---------

Signed-off-by: nithinraok <[email protected]>
Signed-off-by: nithinraok <[email protected]>
Co-authored-by: nithinraok <[email protected]>
Signed-off-by: Charlie Truong <[email protected]>

Author: nithinraok