fix xdoctest

wilhelm-lab · Sep 13, 2024 · 3490afa · 3490afa
1 parent 9c24d1c
commit 3490afa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 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 = []

diff --git a/oktoberfest/runner.py b/oktoberfest/runner.py
@@ -39,7 +39,6 @@ def _make_predictions_error_callback(failure_progress_tracker, failure_lock, err
 
 
 def _make_predictions(config, queue_out, progress, lock, batch_df):
-    print("depp")
     predictors = {model_key: pr.Predictor.from_config(config, model_type=model_key) for model_key in config.models}
     predictions = {
         output_name: output