From bee913fd85e5a648a18de8b7e82b4cba48aa458a Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 10 Mar 2022 14:45:45 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- tests/loggers/test_mlflow.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/loggers/test_mlflow.py b/tests/loggers/test_mlflow.py index 6285b1b9913f7..d26424c6ace43 100644 --- a/tests/loggers/test_mlflow.py +++ b/tests/loggers/test_mlflow.py @@ -247,6 +247,7 @@ class CustomModel(BoringModel): def training_step(self, batch, batch_idx): super().training_step(batch, batch_idx) raise BaseException + model = CustomModel() logger = MLFlowLogger("test") run = MagicMock() @@ -256,4 +257,4 @@ def training_step(self, batch, batch_idx): with pytest.raises(BaseException): trainer.fit(model) - client.return_value.set_terminated.assert_called_once_with(logger.run_id, "FAILED") \ No newline at end of file + client.return_value.set_terminated.assert_called_once_with(logger.run_id, "FAILED")