add jittering of timed dataframe

OpenCOMPES · Sep 23, 2023 · 20c5c62 · 20c5c62
1 parent a397ebf
commit 20c5c62
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/sed/core/processor.py b/sed/core/processor.py
@@ -1234,6 +1234,12 @@ def add_jitter(self, cols: Sequence[str] = None):
             cols=cols,
             cols_jittered=cols,
         )
+        if self._timed_dataframe is not None:
+            self._timed_dataframe = self._timed_dataframe.map_partitions(
+                apply_jitter,
+                cols=cols,
+                cols_jittered=cols,
+            )
         metadata = []
         for col in cols:
             metadata.append(col)