fix numpy bug in imputation example

fixes issue Imputation Training Example Err #2934 I filed.