Fix mypy issue

awslabs · May 26, 2023 · 0e859d3 · 0e859d3
1 parent 0aa8e60
commit 0e859d3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/gluonts/torch/model/deepar/module.py b/src/gluonts/torch/model/deepar/module.py
@@ -227,7 +227,10 @@ def prepare_rnn_input(
         observed_input = observed_context
         future_length = future_time_feat.shape[-2]
         if future_length > 1:
-            assert future_target is not None
+            assert (
+                future_target is not None
+                and future_observed_values is not None
+            )
             input = torch.cat(
                 (input, future_target[..., : future_length - 1] / scale),
                 dim=-1,