tf.contrib.factorization.WALSMatrixFactorization( AttributeError: 'module' object has no attribute 'WALSMatrixFactorization')

I just run walsmodel through `gcloud ml-engine jobs`
I run it through my mac terminal, I didn't change any code over here https://github.com/GoogleCloudPlatform/training-data-analyst/tree/master/courses/machine_learning/deepdive/10_recommend/walsmodel

But it always has the following error:
experiment_fn tf.contrib.factorization.WALSMatrixFactorization( AttributeError: 'module' object has no attribute 'WALSMatrixFactorization'

`gcloud ml-engine jobs submit training wals_190721_012226 --region=us-east1 --module-name=walsmodel.task --runtime-version 1.14 --python-version 3.5 --package-path=/home/mabodx/anguis/news_recommendation/10_recommend/walsmodel --job-dir=gs://buzzbreak/news_recommendation_2019_07_21T01_18_02Z/ --staging-bucket=gs://buzzbreak --scale-tier=BASIC_GPU --runtime-version= -- --output_dir=gs://buzzbreak/news_recommendation_2019_07_21T01_18_02Z/model_trained_190721_012212 --input_path=gs://buzzbreak/news_recommendation_2019_07_21T01_18_02Z/news/data --num_epochs=500 --nitems=39681 --nusers=38781 --topk=1000`