fixed tf2 exposing python functionality