How to use the pyuvsim.utils.iter_array_split function in pyuvsim
To help you get started, we’ve selected a few pyuvsim examples, based on popular ways it is used in public projects.
Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
RadioAstronomySoftwareGroup / pyuvsim / pyuvsim / uvsim.py View on Github 
- Within the task loop, decide on source chunks and make skymodels on the fly.
"""
Nbltf = Nbls * Ntimes * Nfreqs
split_srcs = False
if (Nbltf < Npus) and (Npus < Nsrcs):
split_srcs = True
if split_srcs:
src_inds, Nsrcs_local = simutils.iter_array_split(rank, Nsrcs, Npus)
task_inds = range(Nbltf)
Ntasks_local = Nbltf
else:
task_inds, Ntasks_local = simutils.iter_array_split(rank, Nbltf, Npus)
src_inds = range(Nsrcs)
Nsrcs_local = Nsrcs
return task_inds, src_inds, Ntasks_local, Nsrcs_local
pyuvsim
A comprehensive simulation package for radio interferometers in python
Package Health Score
75 / 100Popular pyuvsim functions
- pyuvsim.analyticbeam.AnalyticBeam
- pyuvsim.astropy_interface.MoonLocation
- pyuvsim.astropy_interface.MoonLocation.from_selenocentric
- pyuvsim.astropy_interface.SkyCoord
- pyuvsim.astropy_interface.Time
- pyuvsim.data.DATA_PATH
- pyuvsim.mpi
- pyuvsim.mpi.shared_mem_bcast
- pyuvsim.mpi.world_comm.Barrier
- pyuvsim.mpi.world_comm.bcast
- pyuvsim.profiling.prof
- pyuvsim.profiling.set_profiler
- pyuvsim.simsetup
- pyuvsim.simsetup.initialize_uvdata_from_params
- pyuvsim.simsetup.SkyModelData
- pyuvsim.spherical_coordinates_basis_transformation.r_hat
- pyuvsim.telescope.BeamList
- pyuvsim.utils
- pyuvsim.utils.check_file_exists_and_increment
- pyuvsim.utils.iter_array_split