Add `mode` parameter:

`.. py:function:: set_calc_mode(mode)`