From 463db377459597c4e8924252d608ed3ed83abe80 Mon Sep 17 00:00:00 2001 From: Polina Turishcheva Date: Fri, 8 Mar 2024 14:53:05 +0100 Subject: [PATCH] fix_shift_replacement --- neuralpredictors/layers/encoders/firing_rate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/neuralpredictors/layers/encoders/firing_rate.py b/neuralpredictors/layers/encoders/firing_rate.py index 2c84ff99..f4250ecc 100644 --- a/neuralpredictors/layers/encoders/firing_rate.py +++ b/neuralpredictors/layers/encoders/firing_rate.py @@ -67,7 +67,8 @@ def forward( if detach_core: x = x.detach() - if self.shifter: + if self.shifter and shift is None: + # if shift is defined - no need to change it if pupil_center is None: raise ValueError("pupil_center is not given") shift = self.shifter[data_key](pupil_center, trial_idx)