fix: return value for regression line casted to float (#115)

radicalbit · Jul 17, 2024 · d5d862d · d5d862d
1 parent c2f5437
commit d5d862d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):