Skip to content

Commit

Permalink
Merge pull request #252 from LSSTDESC/parallel_label
Browse files Browse the repository at this point in the history
Label non-parallel stages as such
  • Loading branch information
joezuntz authored Jun 16, 2022
2 parents ba79270 + e9656cd commit ba13f68
Show file tree
Hide file tree
Showing 18 changed files with 31 additions and 11 deletions.
1 change: 1 addition & 0 deletions txpipe/auxiliary_maps.py
Original file line number Diff line number Diff line change
Expand Up @@ -308,6 +308,7 @@ class TXUniformDepthMap(PipelineStage):
make aux_lens_maps do, so may not always be useful.
"""
name = "TXUniformDepthMap"
parallel = False
# make a mask from the auxiliary maps
inputs = [("mask", MapsFile)]
outputs = [("aux_lens_maps", MapsFile)]
Expand Down
2 changes: 2 additions & 0 deletions txpipe/blinding.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@ class TXBlinding(PipelineStage):
"""

name = "TXBlinding"
parallel = False
inputs = [
("twopoint_data_real_raw", SACCFile),
]
Expand Down Expand Up @@ -235,6 +236,7 @@ class TXNullBlinding(PipelineStage):
"""

name = "TXNullBlinding"
parallel = False
inputs = [
("twopoint_data_real_raw", SACCFile),
]
Expand Down
2 changes: 2 additions & 0 deletions txpipe/convergence.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ class TXConvergenceMaps(PipelineStage):
This uses the wlmassmap library, which is included as a submodule in TXPipe.
"""
name = "TXConvergenceMaps"
parallel = False
inputs = [
("source_maps", MapsFile),
]
Expand Down Expand Up @@ -89,6 +90,7 @@ class TXConvergenceMapPlots(PipelineStage):
"""

name = "TXConvergenceMapPlots"
parallel = False

inputs = [
("convergence_maps", MapsFile),
Expand Down
2 changes: 2 additions & 0 deletions txpipe/covariance.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ class TXFourierGaussianCovariance(PipelineStage):
measured.
"""
name = "TXFourierGaussianCovariance"
parallel = False
do_xi = False

inputs = [
Expand Down Expand Up @@ -581,6 +582,7 @@ class TXRealGaussianCovariance(TXFourierGaussianCovariance):
TJPCov and a fiducial cosmology.
"""
name = "TXRealGaussianCovariance"
parallel = False
do_xi = True

inputs = [
Expand Down
2 changes: 1 addition & 1 deletion txpipe/exposure_info.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ class TXExposureInfo(PipelineStage):
"""

name = "TXExposureInfo"

parallel = False
inputs = []
outputs = [
("exposures", HDFFile),
Expand Down
6 changes: 3 additions & 3 deletions txpipe/input_cats.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ class TXCosmoDC2Mock(PipelineStage):
"""

name = "TXCosmoDC2Mock"

parallel = False
inputs = [("response_model", HDFFile)]

outputs = [
Expand Down Expand Up @@ -958,7 +958,7 @@ class TXBuzzardMock(TXCosmoDC2Mock):
"""

name = "TXBuzzardMock"

parallel = False
inputs = [("response_model", HDFFile)]

outputs = [
Expand Down Expand Up @@ -992,7 +992,7 @@ class TXGaussianSimsMock(TXCosmoDC2Mock):
"""

name = "TXGaussianSimsMock"

parallel = False
inputs = [("response_model", HDFFile)]

outputs = [
Expand Down
1 change: 1 addition & 0 deletions txpipe/jackknife.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ class TXJackknifeCenters(PipelineStage):
"""

name = "TXJackknifeCenters"
parallel = False

inputs = [
("random_cats", RandomsCatalog),
Expand Down
1 change: 1 addition & 0 deletions txpipe/map_correlations.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@ class TXMapCorrelations(PipelineStage):
is a directory.
"""
name = "TXMapCorrelations"
parallel = False
inputs = [
("lens_maps", MapsFile),
("convergence_maps", MapsFile),
Expand Down
2 changes: 1 addition & 1 deletion txpipe/map_plots.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ class TXMapPlots(PipelineStage):
"""

name = "TXMapPlots"

parallel = False
inputs = [
("source_maps", MapsFile),
("lens_maps", MapsFile),
Expand Down
1 change: 1 addition & 0 deletions txpipe/maps.py
Original file line number Diff line number Diff line change
Expand Up @@ -530,6 +530,7 @@ class TXDensityMaps(PipelineStage):
"""

name = "TXDensityMaps"
parallel = False
inputs = [
("lens_maps", MapsFile),
("mask", MapsFile),
Expand Down
1 change: 1 addition & 0 deletions txpipe/masks.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ class TXSimpleMask(PipelineStage):
"""
name = "TXSimpleMask"
parallel = False
# make a mask from the auxiliary maps
inputs = [("aux_lens_maps", MapsFile)]
outputs = [("mask", MapsFile)]
Expand Down
3 changes: 2 additions & 1 deletion txpipe/metacal_gcr_input.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ class TXMetacalGCRInput(PipelineStage):
"""

name = "TXMetacalGCRInput"

parallel = False
inputs = []

outputs = [
Expand Down Expand Up @@ -192,6 +192,7 @@ class TXIngestStars(PipelineStage):
in PSF estimation.
"""
name = "TXIngestStars"
parallel = False
inputs = []

outputs = [
Expand Down
1 change: 1 addition & 0 deletions txpipe/metadata.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@ class TXTracerMetadata(PipelineStage):
"""

name = "TXTracerMetadata"
parallel = False

inputs = [
("shear_catalog", ShearCatalog),
Expand Down
2 changes: 1 addition & 1 deletion txpipe/photoz_mlz.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ class PZPDFMLZ(PipelineStage):
"""

name = "PZPDFMLZ"

parallel = False
inputs = [
("photometry_catalog", HDFFile),
("photoz_trained_model", DataFile),
Expand Down
2 changes: 1 addition & 1 deletion txpipe/photoz_stack.py
Original file line number Diff line number Diff line change
Expand Up @@ -325,7 +325,7 @@ class TXPhotozPlots(PipelineStage):
"""
Make n(z) plots of source and lens galaxies
"""

parallel = False
name = "TXPhotozPlots"
inputs = [("shear_photoz_stack", NOfZFile), ("lens_photoz_stack", NOfZFile)]
outputs = [
Expand Down
9 changes: 6 additions & 3 deletions txpipe/psf_diagnostics.py
Original file line number Diff line number Diff line change
Expand Up @@ -183,7 +183,7 @@ class TXRoweStatistics(PipelineStage):
"""

name = "TXRoweStatistics"

parallel = False
inputs = [("star_catalog", HDFFile)]
outputs = [
("rowe134", PNGFile),
Expand Down Expand Up @@ -340,7 +340,7 @@ def save_stats(self, rowe_stats):
f.close()


class TXStarShearTests(PipelineStage):
class TXGalaxyStarShear(PipelineStage):
"""
Compute and plot star x galaxy and star x star correlations.
Expand All @@ -349,6 +349,7 @@ class TXStarShearTests(PipelineStage):
"""

name = "TXGalaxyStarShear"
parallel = False

inputs = [
("shear_catalog", ShearCatalog),
Expand Down Expand Up @@ -615,7 +616,7 @@ def save_stats(self, galaxy_star_stats, star_star_stats):
f.close()


class TXStarDensityTests(PipelineStage):
class TXGalaxyStarDensity(PipelineStage):
"""
Compute and plot star x galaxy and star x star density correlations
Expand All @@ -624,6 +625,7 @@ class TXStarDensityTests(PipelineStage):
"""

name = "TXGalaxyStarDensity"
parallel = False

inputs = [
("shear_catalog", ShearCatalog),
Expand Down Expand Up @@ -824,6 +826,7 @@ class TXBrighterFatterPlot(PipelineStage):
"""

name = "TXBrighterFatterPlot"
parallel = False

inputs = [("star_catalog", HDFFile)]

Expand Down
2 changes: 2 additions & 0 deletions txpipe/theory.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ class TXTwoPointTheoryReal(PipelineStage):
"""

name = "TXTwoPointTheoryReal"
parallel = False
inputs = [
("twopoint_data_real", SACCFile),
("fiducial_cosmology", FiducialCosmology), # For example lines
Expand Down Expand Up @@ -192,6 +193,7 @@ class TXTwoPointTheoryFourier(TXTwoPointTheoryReal):
"""

name = "TXTwoPointTheoryFourier"
parallel = False
inputs = [
("twopoint_data_fourier", SACCFile),
("fiducial_cosmology", FiducialCosmology), # For example lines
Expand Down
2 changes: 2 additions & 0 deletions txpipe/twopoint_plots.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ class TXTwoPointPlots(PipelineStage):
"""

name = "TXTwoPointPlots"
parallel = False
inputs = [
("twopoint_data_real", SACCFile),
("fiducial_cosmology", FiducialCosmology), # For example lines
Expand Down Expand Up @@ -545,6 +546,7 @@ class TXTwoPointPlotsFourier(PipelineStage):
"""

name = "TXTwoPointPlotsFourier"
parallel = False
inputs = [
("summary_statistics_fourier", SACCFile),
("fiducial_cosmology", FiducialCosmology), # For example lines
Expand Down

0 comments on commit ba13f68

Please sign in to comment.