style(beamtransfer): blacken

radiocosmology · Jan 8, 2025 · 1c4a115 · 1c4a115
1 parent 5c88d6d
commit 1c4a115
Showing 1 changed file with 15 additions and 7 deletions.
diff --git a/drift/core/beamtransfer.py b/drift/core/beamtransfer.py
@@ -1740,15 +1740,15 @@ class BeamTransferNoSVD(BeamTransfer):
     to be set. (In this case, the value of `skip_svd` is ignored,
     because the SVD step is automatically skipped.)
     """
-    
+
     # No SV cut
     svcut = 0.0
 
     noise_weight = False
 
     def _svd_num(self, mi):
         """Compute number of SVD modes meeting the cut.
-        
+
         Parameters
         ----------
         mi : integer
@@ -1882,7 +1882,9 @@ def project_vector_svd_to_sky(self, mi, vec, temponly=False, conj=False):
         """
 
         if temponly:
-            raise NotImplementedError("temponly not implemented for no-SVD project_vector_svd_to_sky!")
+            raise NotImplementedError(
+                "temponly not implemented for no-SVD project_vector_svd_to_sky!"
+            )
 
         # Create the output matrix
         svec = np.zeros(
@@ -1897,17 +1899,23 @@ def project_vector_svd_to_sky(self, mi, vec, temponly=False, conj=False):
 
             # Loop through frequencies, doing tel-to-sky projection at each freq
             for fi in range(self.nfreq):
-                svec[fi] = np.dot(beam[fi].T.conj(), vec.reshape(self.nfreq, self.ntel, -1)[fi]).reshape(
-                    (self.telescope.num_pol_sky, self.telescope.lmax + 1) + vec.shape[1:]
+                svec[fi] = np.dot(
+                    beam[fi].T.conj(), vec.reshape(self.nfreq, self.ntel, -1)[fi]
+                ).reshape(
+                    (self.telescope.num_pol_sky, self.telescope.lmax + 1)
+                    + vec.shape[1:]
                 )
 
         else:
             ibeam = self.invbeam_m(mi).reshape((self.nfreq, self.nsky, self.ntel))
 
             # Loop through frequencies, doing tel-to-sky projection at each freq
             for fi in range(self.nfreq):
-                svec[fi] = np.dot(ibeam[fi], vec.reshape(self.nfreq, self.ntel, -1)[fi]).reshape(
-                    (self.telescope.num_pol_sky, self.telescope.lmax + 1) + vec.shape[1:]
+                svec[fi] = np.dot(
+                    ibeam[fi], vec.reshape(self.nfreq, self.ntel, -1)[fi]
+                ).reshape(
+                    (self.telescope.num_pol_sky, self.telescope.lmax + 1)
+                    + vec.shape[1:]
                 )
 
         return svec