* change default stop_label to None

* correct base where units was used accidentally instead of the generic input_data
* add route_psth, which decides between a standard PSTHWidget and a PSTHWidget inside a TimeIntervalsSelector wrapper
* change view to use new routing function

nwbwidgets/base.py

@@ -396,7 +396,7 @@ def __init__(self, input_data, **kwargs):
 
         trials = list(self.intervals_tables.values())[0]
         inner_widget = self.InnerWidget(
-            units=self.input_data,
+            input_data=self.input_data,
             trials=trials, 
             **kwargs
         )

nwbwidgets/misc.py

@@ -12,6 +12,7 @@
 from pynwb.misc import AnnotationSeries, Units, DecompositionSeries
 
 from .analysis.spikes import compute_smoothed_firing_rate
+from .base import TimeIntervalsSelector
 from .controllers import (
     make_trial_event_controller,
     GroupAndSortController,
@@ -283,9 +284,6 @@ def __init__(
 
         if trials is None:
             self.trials = self.get_trials()
-            if self.trials is None:
-                self.children = [widgets.HTML("No trials present")]
-                return
         else:
             self.trials = trials
 
@@ -535,6 +533,16 @@ def update(
         fig.subplots_adjust(wspace=0.3)
         return fig
 
+class IntervalsPSTHWidget(TimeIntervalsSelector):
+    InnerWidget = PSTHWidget
+
+def route_psth(units, **kwargs):
+    trials = units.get_ancestor("NWBFile").trials
+    if trials is None:
+        return IntervalsPSTHWidget(units, **kwargs)
+    else:
+        return PSTHWidget(units, **kwargs)
+
 
 def show_histogram(
         data, ax: plt.Axes, start: float, end: float, group_inds=None, nbins: int = 30

nwbwidgets/utils/units.py

@@ -116,7 +116,7 @@ def align_by_time_intervals(
     index,
     intervals,
     start_label="start_time",
-    stop_label="stop_time",
+    stop_label=None,
     start=0.0,
     end=0.0,
     rows_select=(),

nwbwidgets/view.py

@@ -39,7 +39,7 @@ def show_dynamic_table(node, **kwargs) -> widgets.Widget:
         {
             "Summary": DynamicTableSummaryWidget,
             "Session Raster": misc.RasterWidget,
-            "Grouped PSTH": misc.PSTHWidget,
+            "Grouped PSTH": misc.route_psth,
             "Raster Grid": misc.RasterGridWidget,
             "Tuning Curves": misc.TuningCurveWidget,
             "Combined": misc.TuningCurveExtendedWidget,