diff --git a/src/vai_lab/DataProcessing/plugins/reinvent.py b/src/vai_lab/DataProcessing/plugins/reinvent.py index 15274be6..3f7d74fa 100644 --- a/src/vai_lab/DataProcessing/plugins/reinvent.py +++ b/src/vai_lab/DataProcessing/plugins/reinvent.py @@ -71,14 +71,11 @@ def __init__(self, config = {}, data_in = [None], ini = False): class ReinventModel: - def __init__(self, seed=1, acquisition="uncertainty"): - self.seed = seed - self.acquisition = acquisition + def __init__(self): return - def do_run(self): - seed = self.seed - acquisition = self.acquisition + def do_run(self, seed=1, acquisition="uncertainty"): + ############################## # Quick options FIT_MODEL = True # whether to fit a Stan model or not