Update self._queue in every partial_fit call

uxlfoundation · Aug 30, 2024 · 4bc41d0 · 4bc41d0
1 parent b10b58a
commit 4bc41d0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/onedal/basic_statistics/incremental_basic_statistics.py b/onedal/basic_statistics/incremental_basic_statistics.py
@@ -92,8 +92,7 @@ def partial_fit(self, X, weights=None, queue=None):
         self : object
             Returns the instance itself.
         """
-        if not hasattr(self, "_queue"):
-            self._queue = queue
+        self._queue = queue
         policy = self._get_policy(queue, X)
         X, weights = _convert_to_supported(policy, X, weights)
 

diff --git a/onedal/spmd/basic_statistics/incremental_basic_statistics.py b/onedal/spmd/basic_statistics/incremental_basic_statistics.py
@@ -48,8 +48,7 @@ def partial_fit(self, X, weights=None, queue=None):
         self : object
             Returns the instance itself.
         """
-        if not hasattr(self, "_queue"):
-            self._queue = queue
+        self._queue = queue
         policy = super(IncrementalBasicStatistics_base, self)._get_policy(queue, X)
         X, weights = _convert_to_supported(policy, X, weights)