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

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