fix assert

zm711 · Oct 4, 2023 · ce38e36 · ce38e36
1 parent 1e2dceb
commit ce38e36
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/spikeanalysis/spike_plotter.py b/src/spikeanalysis/spike_plotter.py
@@ -72,7 +72,7 @@ def set_analysis(self, analysis: SpikeAnalysis):
             The SpikeAnalysis object for plotting
 
         """
-        assert isinstance(analysis, SpikeAnalysis), "analysis must be a SpikeAnalysis dataset"
+        assert isinstance(analysis, (SpikeAnalysis, CuratedSpikeAnalysis)), "analysis must be a SpikeAnalysis dataset"
         self.data = analysis
 
     def plot_zscores(