FIX Access to attributes of individual NB objects in dask NB

rapidsai · Nov 17, 2020 · fb2aca3 · fb2aca3
1 parent 238a8de
commit fb2aca3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/cuml/dask/naive_bayes/naive_bayes.py b/python/cuml/dask/naive_bayes/naive_bayes.py
@@ -136,8 +136,8 @@ def _merge_counts_to_model(models):
         modela = first(models)
 
         for model in models[1:]:
-            modela._feature_count_ += model._feature_count_
-            modela._class_count_ += model._class_count_
+            modela.feature_count_ += model.feature_count_
+            modela.class_count_ += model.class_count_
         return modela
 
     @staticmethod