suppress dvc logs (#583)

Co-authored-by: David de la Iglesia Castro <[email protected]>

Author: Dave Berenbaum

src/dvclive/live.py

@@ -134,6 +134,10 @@ def _init_dvc(self):
                 self._save_dvc_exp = False
 
         self._dvc_repo = get_dvc_repo()
+
+        dvc_logger = logging.getLogger("dvc")
+        dvc_logger.setLevel(os.getenv(env.DVCLIVE_LOGLEVEL, "WARNING").upper())
+
         if (self._dvc_repo is None) or isinstance(self._dvc_repo.scm, NoSCM):
             if self._save_dvc_exp:
                 logger.warning(

tests/test_main.py

@@ -1,4 +1,5 @@
 import json
+import logging
 import os
 
 import pytest
@@ -484,3 +485,8 @@ def test_make_dvcyaml(tmp_dir, dvcyaml):
     dvcyaml_path = tmp_dir / dvclive.dir / "dvc.yaml"
 
     assert dvcyaml_path.is_file()
+
+
+def test_suppress_dvc_logs(tmp_dir, mocked_dvc_repo):
+    Live()
+    assert logging.getLogger("dvc").level == 30