Fix typo

awslabs · Jan 10, 2024 · 9804eac · 9804eac
1 parent cd63c2a
commit 9804eac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gluonts/model/forecast_generator.py b/src/gluonts/model/forecast_generator.py
@@ -116,7 +116,7 @@ def __call__(
         for batch in inference_data_loader:
             inputs = select(input_names, batch, ignore_missing=True)
             (outputs,), loc, scale = prediction_net(*inputs.values())
-            outputs = to_numpy(output)
+            outputs = to_numpy(outputs)
             if scale is not None:
                 outputs = outputs * to_numpy(scale[..., None])
             if loc is not None: