From d5d862deb02354112cafa15552ffc7973d451452 Mon Sep 17 00:00:00 2001
From: Stefano Zamboni <39366866+SteZamboni@users.noreply.github.com>
Date: Wed, 17 Jul 2024 09:10:24 +0200
Subject: [PATCH] fix: return value for regression line casted to float (#115)

---
 spark/jobs/metrics/model_quality_regression_calculator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spark/jobs/metrics/model_quality_regression_calculator.py b/spark/jobs/metrics/model_quality_regression_calculator.py
index b15a759f..14343851 100644
--- a/spark/jobs/metrics/model_quality_regression_calculator.py
+++ b/spark/jobs/metrics/model_quality_regression_calculator.py
@@ -203,7 +203,7 @@ def get_regression_line(model: ModelOut, dataframe: DataFrame):
         c = lr_model.coefficients[0]
         i = lr_model.intercept
 
-        return [[0, i], [1, c + i]]
+        return [[0, float(i)], [1, float(c + i)]]
 
     @staticmethod
     def residual_metrics(model: ModelOut, dataframe: DataFrame):