lightning: Update test.

daavoo · daavoo · commit 096f0b79379a · 2022-08-23T12:39:17.000+02:00
Per Lightning-AI/pytorch-lightning#9754
diff --git a/tests/test_lightning.py b/tests/test_lightning.py
@@ -86,7 +86,7 @@ def test_lightning_integration(tmp_dir):
     # init logger
     dvclive_logger = DvcLiveLogger("test_run", path="logs")
     trainer = Trainer(
-        logger=dvclive_logger, max_epochs=1, checkpoint_callback=False
+        logger=dvclive_logger, max_epochs=1, enable_checkpointing=False
     )
     trainer.fit(model)
 

Original file line number	Diff line number	Diff line change
`@@ -86,7 +86,7 @@ def test_lightning_integration(tmp_dir):`
`86`	`86`	`# init logger`
`87`	`87`	`dvclive_logger = DvcLiveLogger("test_run", path="logs")`
`88`	`88`	`trainer = Trainer(`
`89`		`- logger=dvclive_logger, max_epochs=1, checkpoint_callback=False`
	`89`	`+ logger=dvclive_logger, max_epochs=1, enable_checkpointing=False`
`90`	`90`	`)`
`91`	`91`	`trainer.fit(model)`
`92`	`92`