diff --git a/mokapot/brew.py b/mokapot/brew.py index 8dc7f73..e415bd3 100644 --- a/mokapot/brew.py +++ b/mokapot/brew.py @@ -149,6 +149,7 @@ def brew( ) except TypeError: + LOGGER.info("No idea how we ended up here, but this seems to be standard training oO?! tf.") train_sets = list( make_train_sets( test_idx=test_folds_idx, @@ -539,6 +540,8 @@ def _fit_model(train_set, psms, model, fold): try: model.fit(train_set) except RuntimeError as msg: + LOGGER.info("=== Analyzing Fold %i ===", fold + 1) + LOGGER.info(msg) if str(msg) != "Model performs worse after training.": raise diff --git a/mokapot/mokapot.py b/mokapot/mokapot.py index 9f58f8e..3c76e8a 100644 --- a/mokapot/mokapot.py +++ b/mokapot/mokapot.py @@ -120,10 +120,6 @@ def main(main_args=None): rng=config.seed, ) logging.info("") - logging.info("BREW DELETE ME") - for s in scores: - logging.info(f"{s}") - logging.info(f"{(s == 0).all()}") if config.dest_dir is not None: config.dest_dir.mkdir(exist_ok=True)