Add type error suppressions for upcoming upgrade

Differential Revision: D66831419 fbshipit-source-id: 5fc0474774d138f157f776c4071a1917c12a5cf7
facebookresearch · Dec 6, 2024 · 53db30a · 53db30a
1 parent a8f7a48
commit 53db30a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/kats/models/globalmodel/model.py b/kats/models/globalmodel/model.py
@@ -1120,6 +1120,7 @@ def _single_pass_s2s(
             self._valid_tensor(tmp_encode, "encoder_output")
 
             if training_mode or is_valid:
+                # pyre-fixme[29]: `Union[Module, Tensor]` is not a function.
                 encoder.prepare_decoder(decoder)
                 encoder_step = (
                     batch.training_encoder_step_num