def _set_coef_prior_idx()

in orbit/template/ktr.py [0:0]

• 8 lines of code
• 5 McCabe index (conditional complexity)

    def _set_coef_prior_idx(self):
        if self._coef_prior_list and len(self._regressor_col) > 0:
            for x in self._coef_prior_list:
                prior_regressor_col_idx = [
                    np.where(np.array(self._regressor_col) == col)[0][0]
                    for col in x[KTRTimePointPriorKeys.PRIOR_REGRESSOR_COL.value]
                ]
                x.update({"prior_regressor_col_idx": prior_regressor_col_idx})