diff --git a/src/rail/creation/noisifier.py b/src/rail/creation/noisifier.py index 4042cdad..aa31bae8 100644 --- 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 index 1faaf929..10794cc1 100644 --- 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)]