def _convert_col_types(self, noteInfo: pd.DataFrame) -> pd.DataFrame:
    """Convert pandas types to numpy types.

    This conversion is necessary for scikit-learn compatibility.
    """
    for col, dtype in noteInfo.dtypes.to_dict().items():
      if isinstance(dtype, pd.Int64Dtype):
        assert noteInfo[col].isna().sum() == 0
        noteInfo[col] = noteInfo[col].astype(np.int64)
      if isinstance(dtype, pd.Float64Dtype):
        noteInfo[col] = noteInfo[col].astype(np.float64)
    return noteInfo
Convert pandas types to numpy types.

    This conversion is necessary for scikit-learn compatibility.
    
sourcecode/scoring/pflip_plus_model.py