test: update h2o schema integration test to make sure binary model is…

… used
capitalone · Sep 26, 2024 · c95139e · c95139e
1 parent 385696e
commit c95139e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration/test_schema.py b/tests/integration/test_schema.py
@@ -111,4 +111,4 @@ def test_estimator_h2o_schema_train(
     model_artifact = experiment.artifact(name=schema_cls.__name__)
 
     assert len(project.schema_["parameters"]) == len(experiment.parameters())
-    assert model_artifact.get_data(deserialize="h2o").__class__.__name__ == schema_cls.__name__
+    assert model_artifact.get_data(deserialize="h2o_binary").__class__.__name__ == schema_cls.__name__