Update direct import of TF to its correct location instead of importing directly from the python init file.#267