Hi,
The return statement for compute_gradient() seems to be swapped and should be:
return dj_dw, dj_db
If you look at the test cells it is expecting the order of tmp_dj_dw, tmp_dj_db
Thank you this works