update operate aurora to use KernelDataset

simpeg · Jun 25, 2022 · caae96e · caae96e
1 parent 6609f56
commit caae96e
Show file tree

Hide file tree

Showing 2 changed files with 369 additions and 523 deletions.
diff --git a/aurora/tf_kernel/dataset.py b/aurora/tf_kernel/dataset.py
@@ -101,9 +101,9 @@ def from_run_summary(self, run_summary,
     def add_duration(self):
         """
         """
-        timedeltas = df.end - df.start
+        timedeltas = self.df.end - self.df.start
         durations = [x.seconds for x in timedeltas]
-        df["duration"] = durations
+        self.df["duration"] = durations
         return
 
     def drop_runs_shorter_than(self, duration, units="s"):