def _add_deprecated_columns()

in sourcecode/scoring/run_scoring.py [0:0]

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

def _add_deprecated_columns(scoredNotes: pd.DataFrame) -> pd.DataFrame:
  """Impute columns which are no longer used but must be maintained in output.

  Args:
    scoredNotes: DataFrame containing note scoring output

  Returns:
    scoredNotes augmented to include deprecated columns filled with dummy values
  """
  for column, columnType in c.deprecatedNoteModelOutputTSVColumnsAndTypes:
    assert column not in scoredNotes.columns
    if columnType == np.double:
      scoredNotes[column] = np.nan
    elif columnType == str:
      scoredNotes[column] = ""
    elif columnType == "category":
      scoredNotes[column] = np.nan
    else:
      assert False, f"column type {columnType} unsupported"
  return scoredNotes