diff --git a/onedal/linear_model/incremental_linear_model.py b/onedal/linear_model/incremental_linear_model.py index 93370a4d45..a3a6461f55 100644 --- a/onedal/linear_model/incremental_linear_model.py +++ b/onedal/linear_model/incremental_linear_model.py @@ -74,8 +74,7 @@ def partial_fit(self, X, y, queue=None): """ module = self._get_backend("linear_model", "regression") - if not hasattr(self, "_queue"): - self._queue = queue + self._queue = queue policy = self._get_policy(queue, X) X, y = _convert_to_supported(policy, X, y) @@ -208,8 +207,7 @@ def partial_fit(self, X, y, queue=None): """ module = self._get_backend("linear_model", "regression") - if not hasattr(self, "_queue"): - self._queue = queue + self._queue = queue policy = self._get_policy(queue, X) X, y = _convert_to_supported(policy, X, y) diff --git a/onedal/spmd/linear_model/incremental_linear_model.py b/onedal/spmd/linear_model/incremental_linear_model.py index b18d29b80c..1703e1faf3 100644 --- a/onedal/spmd/linear_model/incremental_linear_model.py +++ b/onedal/spmd/linear_model/incremental_linear_model.py @@ -58,8 +58,7 @@ def partial_fit(self, X, y, queue=None): "linear_model", "regression" ) - if not hasattr(self, "_queue"): - self._queue = queue + self._queue = queue policy = super(IncrementalLinearRegression_base, self)._get_policy(queue, X) X, y = _convert_to_supported(policy, X, y)