diff --git a/python/cuml/cuml/linear_model/logistic_regression_mg.pyx b/python/cuml/cuml/linear_model/logistic_regression_mg.pyx index bacca15107..b43eb40ddd 100644 --- a/python/cuml/cuml/linear_model/logistic_regression_mg.pyx +++ b/python/cuml/cuml/linear_model/logistic_regression_mg.pyx @@ -1,5 +1,5 @@ # -# Copyright (c) 2023-2024, NVIDIA CORPORATION. +# Copyright (c) 2023-2025, NVIDIA CORPORATION. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -387,7 +387,7 @@ class LogisticRegressionMG(MGFitMixin, LogisticRegression): qnpams, self.standardization, self._num_classes, - &objective32, + &objective64, &num_iters) else: assert self.index_dtype == np.int64, f"unsupported index dtype: {self.index_dtype}" @@ -403,7 +403,7 @@ class LogisticRegressionMG(MGFitMixin, LogisticRegression): qnpams, self.standardization, self._num_classes, - &objective32, + &objective64, &num_iters) self.solver_model.objective = objective64