@@ -515,13 +515,13 @@ def plot_fip(fip_df, channel, ax):
515515
516516 if isinstance (channel , list ):
517517 for sub_channel in channel :
518- if sub_channel not in fip_df [' event' ].unique ():
518+ if sub_channel not in fip_df [" event" ].unique ():
519519 raise Exception ("Cannot plot {}, no data" .format (sub_channel ))
520520
521521 color = get_fip_color (sub_channel )
522522 C = fip_df .query ("event == @sub_channel" )
523- ax .plot (C .timestamps .values , C .data .values , color ,label = sub_channel )
524- ax .set_ylabel (' multiple' , fontsize = 12 )
523+ ax .plot (C .timestamps .values , C .data .values , color , label = sub_channel )
524+ ax .set_ylabel (" multiple" , fontsize = 12 )
525525 ax .legend ()
526526 else :
527527 if channel not in fip_df ["event" ].unique ():
@@ -533,6 +533,7 @@ def plot_fip(fip_df, channel, ax):
533533 ax .set_ylabel (channel , fontsize = 12 )
534534 ax .axhline (0 , color = "k" , linewidth = 0.5 , alpha = 0.25 )
535535
536+
536537def get_fip_color (channel ):
537538 """
538539 Gets the color for FIP
0 commit comments