We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 682e65f commit 642dcc9Copy full SHA for 642dcc9
1 file changed
tests/unit/aiplatform/test_metadata_models.py
@@ -157,8 +157,10 @@ def create_model_file(filename):
157
@pytest.fixture
158
def mock_storage_blob_download_xgboost_xgbmodel_file():
159
def create_model_file(filename):
160
+ xgb.XGBClassifier._estimator_type = "classifier"
161
x, y = make_classification()
162
model = xgb.XGBClassifier()
163
+ model._estimator_type = "classifier"
164
model.fit(x, y)
165
model.save_model(filename)
166
@@ -522,6 +524,7 @@ def test_save_model_xgboost_xgbmodel(
522
524
523
525
526
xgb_model = xgb.XGBClassifier()
527
+ xgb_model._estimator_type = "classifier"
528
xgb_model.fit(x, y)
529
530
aiplatform.init(
0 commit comments