Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Mar 10, 2022
1 parent 2bf6244 commit bee913f
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tests/loggers/test_mlflow.py
Expand Up @@ -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()
Expand All @@ -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")
client.return_value.set_terminated.assert_called_once_with(logger.run_id, "FAILED")

0 comments on commit bee913f

Please sign in to comment.