Commit 7bd84605 authored by lukas leufen's avatar lukas leufen

applied bug fix as mentioned in #74

parent 57b2a4b1
Pipeline #31535 passed with stages
in 6 minutes and 18 seconds
......@@ -31,7 +31,7 @@ class OrdinaryLeastSquaredModel:
self.y = data_y
def predict(self, data):
data = sm.add_constant(self.reshape_xarray_to_numpy(data))
data = sm.add_constant(self.reshape_xarray_to_numpy(data), has_constant="add")
return np.atleast_2d(self.model.predict(data))
from src.model_modules.linear_model import OrdinaryLeastSquaredModel
class TestOrdinaryLeastSquareModel:
def test_constant_input_variable(self):
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment