moving read_data tvector to kwargs

indica-mcf · Aug 17, 2023 · 8e2f2d8 · 8e2f2d8
1 parent b1babc4
commit 8e2f2d8
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/indica/workflows/abstract_bayes_workflow.py b/indica/workflows/abstract_bayes_workflow.py
@@ -31,19 +31,19 @@ def __init__(
         self.read_data(self.diagnostics)
         self.setup_models(self.diagnostics)
 
-    def read_test_data(self, diagnostic_transforms: dict):
+    def read_test_data(self, diagnostic_transforms: dict, tstart=None, tend=None, dt=None):
         # Used with phantom data for purposes of tests
         print("Reading fake data")
         self.reader = None
         self.equilibrium = fake_equilibrium(
-            self.tstart,
-            self.tend,
-            self.dt,
+            tstart,
+            tend,
+            dt,
         )
         self.transforms = diagnostic_transforms
         self.data = {}
 
-    def read_data(self, diagnostics: list):
+    def read_data(self, diagnostics: list, tstart=None, tend=None, dt=None):
         self.reader = ReadST40(
             self.pulse, tstart=self.tstart, tend=self.tend, dt=self.dt
         )