infer dimensions for ACF.tau

voytekresearch · Apr 7, 2024 · a1672f1 · a1672f1
1 parent d613aad
commit a1672f1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/timescales/fit/acf.py b/timescales/fit/acf.py
@@ -227,7 +227,11 @@ def fit(self, lags=None, corrs=None, gen_fits=True, gen_components=False, with_c
 
         if gen_fits and not np.isnan(self.params).any():
             self.gen_corrs_fit(gen_components)
-            self.tau = self.params[:, 0]
+
+            if self.params.ndim == 1:
+                self.tau =self.params[0]
+            else:
+                self.tau = self.params[:, 0]
             self.knee_freq = convert_knee(self.tau)
         else:
             self.tau = np.nan