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

This reverts commit be13afc12fb409a3ac53c98172f052f989e6c06e.
Rebase from 'upstream'
Trimmed the branch to only the tf2_py sub directory