change rust function names

Author: ArthurDeclercq

ms2rescore/feature_generators/ms2.py

@@ -7,9 +7,8 @@
 
 from typing import List, Optional
 
-import numpy as np
 from psm_utils import PSMList
-from ms2rescore_rs import batch_ms2_features_from_spectra
+from ms2rescore_rs import ms2_features_from_ms2spectra
 from ms2rescore.feature_generators.base import FeatureGeneratorBase
 
 logger = logging.getLogger(__name__)
@@ -76,7 +75,7 @@ def add_features(self, psm_list: PSMList) -> None:
         proformas = [psm.peptidoform.proforma.split("/")[0] for psm in psm_list]
         seq_lens = [len(psm.peptidoform.sequence) for psm in psm_list]
 
-        feature_dicts = batch_ms2_features_from_spectra(
+        feature_dicts = ms2_features_from_ms2spectra(
             spectra=spectra,
             proformas=proformas,
             seq_lens=seq_lens,

ms2rescore/feature_generators/ms2pip.py

@@ -27,7 +27,7 @@
 from typing import Optional
 
 from ms2pip import process_MS2_spectra
-from ms2rescore_rs import batch_ms2pip_features_numpy
+from ms2rescore_rs import ms2pip_features_from_prediction_peak_arrays
 
 from psm_utils import PSMList
 
@@ -198,7 +198,7 @@ def _calculate_features(self, psm_list, ms2pip_results):
             obs_b.append(r.observed_intensity["b"])
             obs_y.append(r.observed_intensity["y"])
 
-        results = batch_ms2pip_features_numpy(idx, pred_b, pred_y, obs_b, obs_y)
+        results = ms2pip_features_from_prediction_peak_arrays(idx, pred_b, pred_y, obs_b, obs_y)
 
         for psm_index, feats in results:
             if feats: