Fix a coupl eof references to removed APIs.
Fix a coupl eof references to removed APIs.

yebrahim/datalab,craigcitro/datalab,googledatalab/datalab,yebrahim/datalab,craigcitro/datalab,googledatalab/datalab,parthea/datalab,yebrahim/datalab,parthea/datalab,craigcitro/datalab,googledatalab/datalab,googledatalab/datalab,parthea/datalab,parthea/datalab,googledatalab/datalab,craigcitro/datalab,craigcitro/datalab,parthea/datalab,yebrahim/datalab,yebrahim/datalab,parthea/datalab,googledatalab/datalab