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

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