Skip to content

Commit a01f0ab

Browse files
committed
linting
1 parent 0550b0c commit a01f0ab

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

src/aind_dynamic_foraging_basic_analysis/plot/plot_session_scroller.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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+
536537
def get_fip_color(channel):
537538
"""
538539
Gets the color for FIP

0 commit comments

Comments
 (0)