fix correlation bug

zm711 · Oct 4, 2024 · 8d540c3 · 8d540c3
1 parent 66f8406
commit 8d540c3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/spikeanalysis/spike_analysis.py b/src/spikeanalysis/spike_analysis.py
@@ -414,6 +414,10 @@ def get_raw_firing_rate(
                 self.fr_bins[stim] = bins[fr_window_values]
             self.mean_firing_rate = final_fr
 
+    def zscore_data(self, time_bin_ms, bsl_window, z_window, eps):
+
+        self.z_score_data(time_bin_ms=time_bin_ms, bsl_window=bsl_window, z_window=z_window, eps=eps)
+
     def z_score_data(
         self,
         time_bin_ms: Union[list[float], float],
@@ -734,6 +738,7 @@ def trial_correlation(
             data = getattr(self, "psths")
         elif dataset == "raw":
             data = getattr(self, "raw_firing_rate")
+            bins = self.fr_bins
         elif dataset == "z_scores":
             data = getattr(self, "raw_zscores")
             bins = self.z_bins