modified: hera_pspec/uvpspec.py

HERA-Team · May 27, 2019 · d4e7a64 · d4e7a64
1 parent 643cc07
commit d4e7a64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hera_pspec/uvpspec.py b/hera_pspec/uvpspec.py
@@ -1679,7 +1679,7 @@ def generate_noise_spectra(self, spw, polpair, Tsys, blpairs=None,
         for i, blp in enumerate(blpairs):
             # get indices
             inds = self.blpair_to_indices(blp)
-
+            assert isinstance(Tsys[blp], (float, np.float, int, np.int)) or Tsys[blp].shape[0] == self.Ntimes, "Tsys must be a float or an ndarray with shape[0] == Ntimes"
             P_blp = []
             # iterate over time axis
             for j, ind in enumerate(inds):