Merge pull request #153 from LSSTDESC/ceci2

Update for ceci version 2
LSSTDESC · Jul 17, 2024 · c25bd78 · c25bd78
2 parents c9a66b7 + a335e10
commit c25bd78
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/docs/source/new_rail_stage.rst b/docs/source/new_rail_stage.rst
@@ -35,8 +35,6 @@ The following example has all of the required pieces of a ``RailStage`` and almo
    inputs = [('input', PqHandle)]
        outputs = [('output', PqHandle)]
 
-       def __init__(self, args, comm=None):
-           RailStage.__init__(self, args, comm=comm)
 
        def run(self):
            data = self.get_data('input', allow_missing=True)
@@ -101,8 +99,8 @@ Here is an example of a slightly more complicated ``RailStage``.
        outputs = [('output', QPHandle),
                   ('single_NZ', QPHandle)]
 
-       def __init__(self, args, comm=None):
-           PZSummarizer.__init__(self, args, comm=comm)
+       def __init__(self, args, **kwargs):
+           super().__init__(self, args, **kwargs)
            self.zgrid = None
 
        def run(self):