Merge pull request #480 from int-brain-lab/training_plots

Training plots

Author: mayofaulkner

ibllib/pipes/training_status.py

@@ -420,7 +420,7 @@ def plot_performance_easy_median_reaction_time(df, subject):
 
     y2 = {'column': 'combined_reaction_time',
           'title': 'Median reaction time (s)',
-          'lim': [0.1, np.max([10, np.max(df.combined_reaction_time.values)])],
+          'lim': [0.1, np.nanmax([10, np.nanmax(df.combined_reaction_time.values)])],
           'log': True}
     ax = plot_over_days(df, y1, y2, subject)
 
@@ -548,6 +548,9 @@ def make_plots(session_path, one, df=None, save=False, upload=False):
 
     df = df[df['task_protocol'] != 'habituation']
 
+    if len(df) == 0:
+        return
+
     ax1 = plot_trial_count_and_session_duration(df, subject)
     ax2 = plot_performance_easy_median_reaction_time(df, subject)
     ax3 = plot_heatmap_performance_over_days(df, subject)