Merge pull request #234 from pollytur/fix_shift_replacement_if_define…

…d_2d_encoder fix_shift_replacement_if_defined_2d_encoder
sinzlab · Mar 8, 2024 · a57dc39 · a57dc39
2 parents 4ceba76 + 463db37
commit a57dc39
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)