Merge pull request #104 from LSSTDESC/issue/103/change-behavior-of-se…

…ed-param Change behavior of `seed` in `Noisifier` and `Selector`
LSSTDESC · Apr 24, 2024 · eee62e9 · eee62e9
2 parents 5bdb1e3 + cc443d6
commit eee62e9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/rail/creation/noisifier.py b/src/rail/creation/noisifier.py
@@ -6,7 +6,7 @@
 
 from rail.core.stage import RailStage
 from rail.core.data import PqHandle
-
+from ceci.config import StageParameter as Param
 
 class Noisifier(RailStage):
     """Base class Noisifier, which adds noise to the input catalog
@@ -18,7 +18,9 @@ class Noisifier(RailStage):
 
     name = 'Noisifier'
     config_options = RailStage.config_options.copy()
-    config_options.update(seed=1337)
+    config_options.update(
+        seed=Param(default=None, required=False, msg="Set to an `int` to force reproducible results."),
+    )
     inputs = [('input', PqHandle)]
     outputs = [('output', PqHandle)]
 

diff --git a/src/rail/creation/selector.py b/src/rail/creation/selector.py
@@ -22,6 +22,7 @@ class Selector(RailStage):
     config_options = RailStage.config_options.copy()
     config_options.update(
         drop_rows=Param(bool, True, msg="Drop selected rows from output table"),
+        seed=Param(default=None, required=False, msg="Set to an `int` to force reproducible results."),
     )
     inputs = [('input', PqHandle)]
     outputs = [('output', PqHandle)]