From e595c8d2daec7d4ba1efceb688f78c82e577e9a4 Mon Sep 17 00:00:00 2001
From: Edward Yang <ezyang@meta.com>
Date: Wed, 20 Nov 2024 14:56:19 -0800
Subject: [PATCH] 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
---
 kats/models/globalmodel/model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/kats/models/globalmodel/model.py b/kats/models/globalmodel/model.py
index d494b6a3..7d3dad77 100644
--- 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