Update self._queue in every partial_fit call

uxlfoundation · Aug 30, 2024 · 7fda21f · 7fda21f
1 parent 0d13781
commit 7fda21f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git 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
@@ -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)