Changing reinvent class input

AaltoPML · Dec 22, 2023 · 6ddd315 · 6ddd315
1 parent aa4374c
commit 6ddd315
Showing 1 changed file with 3 additions and 6 deletions.
diff --git 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