diff --git a/ibllib/pipes/training_status.py b/ibllib/pipes/training_status.py index 7c98a028a..26bdbf9ab 100644 --- a/ibllib/pipes/training_status.py +++ b/ibllib/pipes/training_status.py @@ -685,6 +685,10 @@ def add_training_lines(df, ax): status = df.drop_duplicates(subset='training_status', keep='first') for _, st in status.iterrows(): + + if st['training_status'] in ['untrainable', 'unbiasable']: + continue + if TRAINING_STATUS[st['training_status']][0] <= 0: continue