Import error: No module named 'psycopg2'

I am getting the following error trying to run the Python 3.6 AWS Lambda

START RequestId: 130fc633-d45c-11e7-b41f-bd31451e0a34 Version: $LATEST
Unable to import module 'handler': No module named 'psycopg2'
END RequestId: 130fc633-d45c-11e7-b41f-bd31451e0a34

Any suggestions on how to fix it? Thanks.

Found the solution here https://github.com/jkehler/awslambda-psycopg2

assuming dockerizePip: true is the best option