fix xdoctest

wilhelm-lab · Sep 13, 2024 · 395373f · 395373f
1 parent 9c24d1c
commit 395373f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/oktoberfest/predict/predictor.py b/oktoberfest/predict/predictor.py
@@ -258,15 +258,15 @@ def predict_in_chunks(self, data: pd.DataFrame, chunk_idx: list[pd.Index], **kwa
             >>>                         "PRECURSOR_CHARGE": [1,2],
             >>>                         "FRAGMENTATION": ["HCD","HCD"],
             >>>                         "PEPTIDE_LENGTH": [8,9]})
-            >>> idx = list(group_iterator(df=meta_df, group_by_column="PEPTIDE_LENGTH"))
             >>> var = Spectra._gen_vars_df()
             >>> library = Spectra(obs=meta_df, var=var)
+            >>> idx = list(group_iterator(df=library.obs, group_by_column="PEPTIDE_LENGTH"))
             >>> intensity_predictor = pr.Predictor.from_koina(
             >>>                         model_name="Prosit_2020_intensity_HCD",
             >>>                         server_url="koina.wilhelmlab.org:443",
             >>>                         ssl=True,
             >>>                         targets=["intensities", "annotation"])
-            >>> predictions = intensity_predictor.predict_in_chunks(data=library, chunk_idx=idx)
+            >>> predictions = intensity_predictor.predict_in_chunks(data=library.obs, chunk_idx=idx)
             >>> print(predictions)
         """
         results = []