Skip to content

Commit

Permalink
style: apply black
Browse files Browse the repository at this point in the history
  • Loading branch information
jrs65 committed Feb 3, 2021
1 parent 47d5bd1 commit 839bbf1
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 6 deletions.
2 changes: 1 addition & 1 deletion draco/analysis/ringmapmaker.py
Original file line number Diff line number Diff line change
Expand Up @@ -512,7 +512,7 @@ def B(phi, u, sigma):
# Get the effective beamwidth for the polarisation combination
sig_a = beam_width[pa](freq, dec)
sig_b = beam_width[pb](freq, dec)
sig = sig_a * sig_b / (sig_a**2 + sig_b**2)**0.5
sig = sig_a * sig_b / (sig_a ** 2 + sig_b ** 2) ** 0.5
sig_arr = sig[np.newaxis, :, np.newaxis]

# Calculate the effective beam transfer function at each declination
Expand Down
9 changes: 4 additions & 5 deletions draco/analysis/transform.py
Original file line number Diff line number Diff line change
Expand Up @@ -218,7 +218,7 @@ def process(self, ss):
attrs_from=ss,
distributed=True,
comm=ss.comm,
**output_kwargs
**output_kwargs,
)

# Add gain dataset.
Expand Down Expand Up @@ -501,12 +501,11 @@ def process(self, sstream):
# marray = _make_marray(sstream.vis[:], mmax)
# marray = mpiarray.MPIArray.wrap(marray[:], axis=freq_axis, comm=sstream.comm)


# Assign the visibilities and weights into the container
#ma.vis[:] = marray
#ma.weight[:] = weight_sum[np.newaxis, np.newaxis, :, :]
# ma.vis[:] = marray
# ma.weight[:] = weight_sum[np.newaxis, np.newaxis, :, :]

#ma.redistribute("m")
# ma.redistribute("m")

return ma

Expand Down

0 comments on commit 839bbf1

Please sign in to comment.