bug fix to splineSG

ibs-lab · Mar 11, 2024 · fd7cecd · fd7cecd
1 parent 4c1b2d6
commit fd7cecd
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 30 deletions.
diff --git a/examples/splineSG_motion_correct.ipynb b/examples/splineSG_motion_correct.ipynb
diff --git a/src/cedalion/sigproc/artifact.py b/src/cedalion/sigproc/artifact.py
@@ -208,7 +208,7 @@ def detect_baselineshift(fNIRSdata: cdt.NDTimeSeries, M: cdt.NDTimeSeries):
 
         sig = channel.copy()
         sig.values = np.ones(channel.shape) # initialize array for baseline shift detection
-        sig.values[np.where(channel_M.values == False)[0]] = 0 # set indices where motion is detected to 0
+        sig.values[np.where(channel_M.values == True)[0]] = 0 # set indices where motion is detected to 0
 
         # find locations where signal goes from clean -> motion -> clean
         temp = sig.diff('time')