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