From c311852b13d6663e8e3e9413ad3ec3afc9ac357d Mon Sep 17 00:00:00 2001
From: Mayo Faulkner <mayo.faulkner@ucl.ac.uk>
Date: Wed, 3 May 2023 09:43:50 +0100
Subject: [PATCH] fix for training line plots

---
 ibllib/pipes/training_status.py | 4 ++++
 1 file changed, 4 insertions(+)

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