lightning: support report kwarg (#369)

* lightning: support report kwarg

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: Dave Berenbaum

src/dvclive/lightning.py

@@ -16,11 +16,12 @@ def __init__(
         experiment=None,
         dir: Optional[str] = None,  # noqa pylint: disable=redefined-builtin
         resume: bool = False,
+        report: Optional[str] = "auto",
     ):
 
         super().__init__()
         self._prefix = prefix
-        self._live_init: Dict[str, Any] = {"resume": resume}
+        self._live_init: Dict[str, Any] = {"resume": resume, "report": report}
         if dir is not None:
             self._live_init["dir"] = dir
         self._experiment = experiment

tests/test_lightning.py

@@ -127,3 +127,19 @@ def test_lightning_default_dir(tmp_dir):
     trainer.fit(model)
 
     assert os.path.exists("dvclive")
+
+
+def test_lightning_kwargs(tmp_dir):
+    model = LitXOR()
+    # Handle kwargs passed to Live.
+    dvclive_logger = DVCLiveLogger(dir="dir", report="md")
+    trainer = Trainer(
+        logger=dvclive_logger,
+        max_epochs=2,
+        enable_checkpointing=False,
+        log_every_n_steps=1,
+    )
+    trainer.fit(model)
+
+    assert os.path.exists("dir")
+    assert os.path.exists("dir/report.md")