Updating the definition of the StageParams to remove dtype and includ…

…e required=False.
LSSTDESC · Apr 24, 2024 · cc443d6 · cc443d6
1 parent 45a9b40
commit cc443d6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/rail/creation/noisifier.py b/src/rail/creation/noisifier.py
@@ -19,7 +19,7 @@ class Noisifier(RailStage):
     name = 'Noisifier'
     config_options = RailStage.config_options.copy()
     config_options.update(
-        seed=Param(int, None, msg="If set, creates reproducible results."),
+        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,7 +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(int, None, msg="If set, creates reproducible results."),
+        seed=Param(default=None, required=False, msg="Set to an `int` to force reproducible results."),
     )
     inputs = [('input', PqHandle)]
     outputs = [('output', PqHandle)]