provide ion mobility argument to predict_ccs

ms2pip/_utils/ion_mobility.py

@@ -9,7 +9,7 @@
 
 
 class IonMobility:
-    """Predict ion mobility using IM²Deep."""
+    """Predict ion mobility using IM2Deep."""
 
     def __init__(self, processes=1) -> None:
         # Lazy import to avoid loading loading heavy dependencies when not needed
@@ -27,6 +27,6 @@ def add_im_predictions(self, psm_list: PSMList) -> None:
         """Add ion mobility predictions to the PSMList."""
         logger.info("Predicting ion mobility...")
         predictions: pd.Series = self.predict_fn(
-            psm_list, write_output=False, n_jobs=self.processes
+            psm_list, write_output=False, n_jobs=self.processes, ion_mobility=True
         )
         psm_list["ion_mobility"] = predictions.values