Skip to content

Commit b0a87dc

Browse files
committed
temp. fix cuml verbosity 2
1 parent cf3823d commit b0a87dc

File tree

3 files changed

+6
-3
lines changed

3 files changed

+6
-3
lines changed

cuml/bench.py

100755100644
Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -100,9 +100,6 @@ def parse_args(parser, size=None, loop_types=(),
100100
This is the same parser that was passed to this function.
101101
'''
102102

103-
import cuml
104-
cuml.common.logger = cuml.common.logger.level_critical
105-
106103
parser.add_argument('-n', '--num-threads', '--core-number', default=-1,
107104
dest='threads', type=int,
108105
help='Number of threads to use')

cuml/kmeans.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,11 @@
88
)
99
import numpy as np
1010
from cuml import KMeans
11+
import cuml
1112
import warnings
1213

14+
cuml.common.logger = cuml.common.logger.level_critical
15+
1316

1417
warnings.filterwarnings('ignore', category=FutureWarning)
1518
parser = argparse.ArgumentParser(description='cuML K-means benchmark')

cuml/log_reg.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,9 @@
77
parse_args, measure_function_time, load_data, print_output, accuracy_score
88
)
99
from cuml import LogisticRegression
10+
import cuml
11+
12+
cuml.common.logger = cuml.common.logger.level_critical
1013

1114
parser = argparse.ArgumentParser(description='cuML logistic '
1215
'regression benchmark')

0 commit comments

Comments
 (0)