diff --git a/src/dvclive/live.py b/src/dvclive/live.py index f9726ed..1584325 100644 --- a/src/dvclive/live.py +++ b/src/dvclive/live.py @@ -272,7 +272,9 @@ def _init_dvc_file(self) -> str: return self._dvcyaml raise InvalidDvcyamlError if self._dvcyaml: - raise InvalidDvcyamlError("DVC yaml path is invalid. Must be a string or a Path object.") + raise InvalidDvcyamlError( + "DVC yaml path is invalid. Must be a string or a Path object." + ) return "dvc.yaml" def _init_dvc_pipeline(self): diff --git a/tests/test_dvc.py b/tests/test_dvc.py index e956323..d54833b 100644 --- a/tests/test_dvc.py +++ b/tests/test_dvc.py @@ -230,4 +230,4 @@ def test_test_mode(tmp_dir, monkeypatch, mocked_dvc_repo, dvcyaml_path): assert live._dvc_file != "dvc.yaml" assert live._save_dvc_exp is False assert not os.path.exists("dir") - assert not os.path.exists(dvcyaml_path) \ No newline at end of file + assert not os.path.exists(dvcyaml_path)