From a2153eba991fec063ccbeeb125934b7af036f8e3 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 9 Sep 2024 10:30:26 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 src/dvclive/live.py | 4 +++-
 tests/test_dvc.py   | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

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)