From abbe46e9e23eef1a133abaca0e34da5cd9ca6c50 Mon Sep 17 00:00:00 2001 From: Lorenzo Stella Date: Thu, 31 Aug 2023 11:10:07 +0200 Subject: [PATCH] fix return type --- src/gluonts/evaluation/metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/gluonts/evaluation/metrics.py b/src/gluonts/evaluation/metrics.py index 69bde27a0f..41a2c97bea 100644 --- a/src/gluonts/evaluation/metrics.py +++ b/src/gluonts/evaluation/metrics.py @@ -87,7 +87,7 @@ def coverage(target: np.ndarray, forecast: np.ndarray) -> float: coverage = mean(Y <= \hat{Y}) """ - return np.mean(target <= forecast) + return float(np.mean(target <= forecast)) def mase(