Fix mis-use of super and change to where aliases live

src/rail/core/stage.py

@@ -288,7 +288,7 @@ def set_data(self, tag, data, path=None, do_read=True):
         if isinstance(data, DataHandle):
             aliased_tag = data.tag
             if tag in self.input_tags():
-                self.config.aliases[tag] = aliased_tag
+                self._aliases[tag] = aliased_tag
                 if data.has_path:
                     self._inputs[tag] = data.path
             arg_data = data.data

src/rail/creation/degraders/addRandom.py

@@ -23,7 +23,7 @@ def __init__(self, args, **kwargs):
 
         Does standard Noisifier initialization
         """
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
 
     def _initNoiseModel(self):  # pragma: no cover
         np.random.seed(self.config.seed)

src/rail/creation/degraders/quantityCut.py

@@ -23,7 +23,7 @@ def __init__(self, args, **kwargs):
         Performs standard Degrader initialization as well as defining the cuts
         to be applied.
         """
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self.cuts = None
         self.set_cuts(self.config["cuts"])
 

src/rail/creation/engine.py

@@ -21,7 +21,7 @@ class Modeler(RailStage):  # pragma: no cover
 
     def __init__(self, args, **kwargs):
         """Initialize Modeler"""
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self.model = None
 
     def fit_model(self):
@@ -58,7 +58,7 @@ class Creator(RailStage):  # pragma: no cover
 
     def __init__(self, args, **kwargs):
         """Initialize Creator"""
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self.model = None
         if not isinstance(args, dict):  # pragma: no cover
             args = vars(args)
@@ -146,7 +146,7 @@ class PosteriorCalculator(RailStage):  # pragma: no cover
 
     def __init__(self, args, **kwargs):
         """Initialize PosteriorCalculator"""
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self.model = None
         if not isinstance(args, dict):  # pragma: no cover
             args = vars(args)

src/rail/estimation/algos/naive_stack.py

@@ -37,7 +37,7 @@ class NaiveStackSummarizer(PZSummarizer):
     outputs = [("output", QPHandle), ("single_NZ", QPHandle)]
 
     def __init__(self, args, **kwargs):
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self.zgrid = None
 
     def run(self):

src/rail/estimation/algos/point_est_hist.py

@@ -38,7 +38,7 @@ class PointEstHistSummarizer(PZSummarizer):
     outputs = [("output", QPHandle), ("single_NZ", QPHandle)]
 
     def __init__(self, args, **kwargs):
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self.zgrid = None
         self.bincents = None
 

src/rail/estimation/algos/random_gauss.py

@@ -47,7 +47,7 @@ class RandomGaussEstimator(CatEstimator):
     def __init__(self, args, **kwargs):
         """Constructor:
         Do CatEstimator specific initialization"""
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self.zgrid = None
 
     def _process_chunk(self, start, end, data, first):

src/rail/estimation/algos/true_nz.py

@@ -29,7 +29,7 @@ class TrueNZHistogrammer(RailStage):
     outputs = [("true_NZ", QPHandle)]
 
     def __init__(self, args, **kwargs):
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self.zgrid = None
         self.bincents = None
 

src/rail/estimation/algos/var_inf.py

@@ -61,7 +61,7 @@ class VarInfStackSummarizer(PZSummarizer):
     outputs = [("output", QPHandle), ("single_NZ", QPHandle)]
 
     def __init__(self, args, **kwargs):
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self.zgrid = None
 
     def run(self):

src/rail/estimation/classifier.py

@@ -25,7 +25,7 @@ class CatClassifier(RailStage):  # pragma: no cover
 
     def __init__(self, args, **kwargs):
         """Initialize Classifier"""
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self._output_handle = None
         self.model = None
         if not isinstance(args, dict):  # pragma: no cover
@@ -116,7 +116,7 @@ def __init__(self, args, **kwargs):
         comm : MPI.Comm, optional
             MPI communicator for parallel processing.
         """
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self._output_handle = None
 
     def classify(self, input_data):

src/rail/estimation/estimator.py

@@ -38,7 +38,7 @@ class CatEstimator(RailStage, PointEstimationMixin):
 
     def __init__(self, args, **kwargs):
         """Initialize Estimator"""
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self._output_handle = None
         self.model = None
 

src/rail/estimation/informer.py

@@ -34,7 +34,7 @@ class CatInformer(RailStage):
 
     def __init__(self, args, **kwargs):
         """Initialize Informer that can inform models for redshift estimation"""
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self.model = None
 
     def inform(self, training_data):
@@ -90,7 +90,7 @@ class PzInformer(RailStage):
 
     def __init__(self, args, **kwargs):
         """Initialize Informer that can inform models for redshift estimation"""
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self.model = None
 
     def inform(self, training_data):

src/rail/estimation/summarizer.py

@@ -139,7 +139,7 @@ class SZPZSummarizer(RailStage):
 
     def __init__(self, args, **kwargs):
         """Initialize Estimator that can sample galaxy data."""
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self.model = None
         # NOTE: open model removed from init, need to put an
         # `open_model` call explicitly in the run method for

src/rail/evaluation/evaluator.py

@@ -80,7 +80,7 @@ class Evaluator(RailStage):  #pylint: disable=too-many-instance-attributes
     metric_base_class = None
 
     def __init__(self, args, **kwargs):
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self._output_handle = None
         self._summary_handle = None
         self._single_distribution_summary_handle = None

src/rail/evaluation/single_evaluator.py

@@ -34,7 +34,7 @@ class SingleEvaluator(Evaluator):  # pylint: disable=too-many-instance-attribute
 
     def __init__(self, args, **kwargs):
         """Initialize Evaluator"""
-        super().__init__(self, args, **kwargs)
+        super().__init__(args, **kwargs)
         self._input_data_type = QPOrTableHandle.PdfOrValue.unknown
         self._truth_data_type = QPOrTableHandle.PdfOrValue.unknown
         self._out_table = {}