Skip to content

Commit b510dfe

Browse files
committed
FIX except known error in unit test
1 parent 47269f7 commit b510dfe

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

test/test_pipeline/test_classification.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
import sklearn.ensemble
1414
import sklearn.svm
1515
from sklearn.utils.testing import assert_array_almost_equal
16+
from xgboost.core import XGBoostError
1617

1718
from ConfigSpace.configuration_space import ConfigurationSpace, \
1819
Configuration
@@ -292,6 +293,13 @@ def _test_configurations(self, configurations_space, make_sparse=False,
292293
print(traceback.format_exc())
293294
print(config)
294295
raise e
296+
except XGBoostError as e:
297+
if "std::bad_alloc" in e.args[0]:
298+
continue
299+
else:
300+
print(traceback.format_exc())
301+
print(config)
302+
raise e
295303

296304
def test_get_hyperparameter_search_space(self):
297305
cs = SimpleClassificationPipeline.get_hyperparameter_search_space()

0 commit comments

Comments
 (0)