Revert "Trimmed the branch to only the tf2_py sub directory"

This reverts commit 70194b427f34b4add61b19b630dac0fca460e501.
Rebase from 'upstream' (no changes)
Trimmed the branch to only the tf2_py sub directory