From ce38e36669a4d5d0ebcc2e0552f653d21e4c1b6f Mon Sep 17 00:00:00 2001
From: zm711 <92116279+zm711@users.noreply.github.com>
Date: Wed, 4 Oct 2023 15:41:28 -0400
Subject: [PATCH] fix assert

---
 src/spikeanalysis/spike_plotter.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/spikeanalysis/spike_plotter.py b/src/spikeanalysis/spike_plotter.py
index 648df30..2f8f837 100644
--- 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(