Fix plotly code

src/aind_dynamic_foraging_basic_analysis/__init__.py

@@ -6,5 +6,8 @@
 from .plot.plot_foraging_session import plot_foraging_session  # noqa: F401
 from .plot.plot_foraging_session_plotly import (  # noqa: F401
     plot_foraging_session_plotly,
+    plot_foraging_session_nwb_plotly,
     plot_session_in_time_plotly,
+    plot_session_in_time_nwb_plotly,
+
 )

src/aind_dynamic_foraging_basic_analysis/metrics/trial_metrics.py

@@ -349,7 +349,7 @@ def get_average_signal_window(
     if output_col is None:
         output_col = (
             f"{data_column}_{channel}_{offsets[0]}_"
-            f"{offsets[1]}_{alignment_event.replace('_in_session','')}"
+            f"{offsets[1]}_{alignment_event.replace('_in_session', '')}"
         )
 
     # copy df_trials, drops na values, sort trial by alignment event

src/aind_dynamic_foraging_basic_analysis/plot/plot_foraging_session.py

@@ -66,7 +66,8 @@ def plot_foraging_session_nwb(nwb, **kwargs):
         0,
         1.05,
         f"{nwb.session_id}\n"
-        f'Total trials {len(nwb.df_trials)}, ignored {np.sum(nwb.df_trials["animal_response"]==2)},'
+        f'Total trials {len(nwb.df_trials)},' +
+        'ignored {np.sum(nwb.df_trials["animal_response"] == 2)},'
         f' left {np.sum(nwb.df_trials["animal_response"] == 0)},'
         f' right {np.sum(nwb.df_trials["animal_response"] == 1)}',
         fontsize=8,