Prepare for "Fix type-safety of torch.nn.Module instances": fbcode/…

…[ijk]* Summary: See D52890934 Reviewed By: r-barnes Differential Revision: D66235329 fbshipit-source-id: 56575053a289023723ecee4e0ee95831bcea4f3f
facebookresearch · Nov 20, 2024 · e595c8d · e595c8d
1 parent 23c24c6
commit e595c8d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/kats/models/globalmodel/model.py b/kats/models/globalmodel/model.py
@@ -1126,6 +1126,7 @@ def _single_pass_s2s(
 
             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