Skip to content

Commit 688f5cf

Browse files
committed
Increased time in regression and epochs for pipeline_score
1 parent 3b011cc commit 688f5cf

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

test/test_api/test_api.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -247,8 +247,8 @@ def test_tabular_regression(openml_name, resampling_strategy, backend):
247247
X_train=X_train, y_train=y_train,
248248
X_test=X_test, y_test=y_test,
249249
optimize_metric='r2',
250-
total_walltime_limit=50,
251-
func_eval_time_limit_secs=10,
250+
total_walltime_limit=150,
251+
func_eval_time_limit_secs=40,
252252
enable_traditional_pipeline=False,
253253
)
254254

test/test_pipeline/test_tabular_classification.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ def test_pipeline_score(self, fit_dictionary_tabular_dummy, fit_dictionary_tabul
9393
"""This test makes sure that the pipeline is able to achieve a decent score on dummy data
9494
given the default configuration"""
9595

96-
fit_dictionary_tabular_dummy['epochs'] = 5
96+
fit_dictionary_tabular_dummy['epochs'] = 15
9797

9898
X = fit_dictionary_tabular_dummy['X_train'].copy()
9999
y = fit_dictionary_tabular_dummy['y_train'].copy()

0 commit comments

Comments
 (0)