Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Apr 23, 2024
1 parent f43a05c commit 8339331
Showing 1 changed file with 12 additions and 2 deletions.
14 changes: 12 additions & 2 deletions hera_sim/visibilities/fftvis.py
Original file line number Diff line number Diff line change
Expand Up @@ -253,9 +253,19 @@ def simulate(self, data_model):
# In certain cases, uvdata.get_antpairs() will return a different order
# than the order in the data_array
if data_model.uvdata.time_axis_faster_than_bls:
antpairs = list(zip(data_model.uvdata.ant_1_array[::data_model.uvdata.Ntimes], data_model.uvdata.ant_2_array[::data_model.uvdata.Ntimes] ))
antpairs = list(
zip(
data_model.uvdata.ant_1_array[:: data_model.uvdata.Ntimes],
data_model.uvdata.ant_2_array[:: data_model.uvdata.Ntimes],
)
)
else:
antpairs = list(zip(data_model.uvdata.ant_1_array[:data_model.uvdata.Nbls], data_model.uvdata.ant_2_array[:data_model.uvdata.Nbls] ))
antpairs = list(
zip(
data_model.uvdata.ant_1_array[: data_model.uvdata.Nbls],
data_model.uvdata.ant_2_array[: data_model.uvdata.Nbls],
)
)

# Get pixelized beams if required
logger.info("Preparing Beams...")
Expand Down

0 comments on commit 8339331

Please sign in to comment.