Skip to content

Commit bac85b5

Browse files
itearslShvets Kirill
authored and
Shvets Kirill
committed
Fix batch.py
1 parent 31c8aad commit bac85b5

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

bench.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -505,7 +505,8 @@ def gen_basic_dict(library, algorithm, stage, params, data, alg_instance=None,
505505
alg_instance_params = dict(alg_instance.get_params())
506506
if ('handle' in alg_instance_params
507507
and 'cuml' in str(alg_instance_params['handle'])):
508-
alg_instance_params['dtype'] = str(alg_instance_params['dtype'])
508+
if 'min_samples_split' in alg_instance_params:
509+
alg_instance_params['dtype'] = str(alg_instance_params['dtype'])
509510
result['algorithm_parameters'].update(alg_instance_params)
510511
if alg_params is not None:
511512
result['algorithm_parameters'].update(alg_params)

0 commit comments

Comments
 (0)