Replace IrisClassificationWithoutValidation with IrisClassification (m…

…lflow#4141) Signed-off-by: harupy <17039389+harupy@users.noreply.github.com>
simonvanbernem · Feb 25, 2021 · 498442f · 498442f
1 parent 3c3fcdd
commit 498442f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/pytorch/test_pytorch_autolog.py b/tests/pytorch/test_pytorch_autolog.py
@@ -48,7 +48,7 @@ def pytorch_model_without_validation():
 @pytest.mark.parametrize("log_models", [True, False])
 def test_pytorch_autolog_log_models_configuration(log_models):
     mlflow.pytorch.autolog(log_models=log_models)
-    model = IrisClassificationWithoutValidation()
+    model = IrisClassification()
     dm = IrisDataModule()
     dm.prepare_data()
     dm.setup(stage="fit")