def _prepare_diligence_init_state()

in sourcecode/scoring/reputation_matrix_factorization/diligence_model.py [0:0]


def _prepare_diligence_init_state(noteInitState, raterInitState):
  if noteInitState is not None:
    noteInitState = noteInitState[
      [c.noteIdKey] + [col for col in noteInitState.columns if "lowDiligence" in col]
    ]
    noteInitState.columns = [
      col.replace("lowDiligence", "internal") for col in noteInitState.columns
    ]
  if raterInitState is not None:
    raterInitState = raterInitState[
      [c.raterParticipantIdKey] + [col for col in raterInitState.columns if "lowDiligence" in col]
    ]
    raterInitState.columns = [
      col.replace("lowDiligence", "internal") for col in raterInitState.columns
    ]
  return noteInitState, raterInitState