Skip to content

Commit

Permalink
Fix lbfgs test verifying log level
Browse files Browse the repository at this point in the history
  • Loading branch information
vyasr committed Jan 2, 2025
1 parent 5b66f08 commit 64f80f9
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions python/cuml/cuml/tests/dask/test_dask_logistic_regression.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright (c) 2019-2024, NVIDIA CORPORATION.
# Copyright (c) 2019-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.
Expand All @@ -21,6 +21,7 @@
from sklearn.datasets import make_classification
from sklearn.linear_model import LogisticRegression as skLR
from cuml.internals.safe_imports import cpu_only_import
from cuml.internals import logger
from cuml.testing.utils import array_equal
from scipy.sparse import csr_matrix, load_npz, save_npz
import random
Expand Down Expand Up @@ -197,8 +198,10 @@ def assert_params(
assert qnpams["fit_intercept"] == fit_intercept
assert qnpams["max_iter"] == max_iter
assert qnpams["linesearch_max_iter"] == linesearch_max_iter
assert (
qnpams["verbose"] == 5 if verbose is True else 4
assert qnpams["verbose"] == (
logger.level_enum.debug
if verbose is True
else logger.level_enum.info
) # cuml Verbosity Levels
assert (
lr.output_type == "input" if output_type is None else output_type
Expand Down

0 comments on commit 64f80f9

Please sign in to comment.