remove value variable

ANNUBS · May 10, 2024 · 2cf6867 · 2cf6867
1 parent 6d800ab
commit 2cf6867
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/annubes/task.py b/annubes/task.py
@@ -471,13 +471,9 @@ def _build_trials_inputs(self) -> NDArray[np.float64]:
                 dtype=np.float32,
             )
             for idx, mod in enumerate(self._modalities):
-                value = (
-                    self._rng.choice(self.stim_intensities, 1)
-                    if (self._modality_seq[n] != "X" and self._modality_seq[n] == mod)
-                    else 0
-                )
+                if self._modality_seq[n] != "X" and self._modality_seq[n] == mod:
+                    x[n][self._phases[n]["input"], idx] = self._rng.choice(self.stim_intensities, 1)
                 x[n][self._phases[n]["fix_time"], idx] = self.fix_intensity
-                x[n][self._phases[n]["input"], idx] = value
             x[n][self._phases[n]["input"], self._n_inputs - 1] = 1  # start cue
             # add noise
             x[n] += noise_factor * self._rng.normal(loc=0, scale=1, size=x[n].shape)