live.logger: Honor DVCLIVE_LOGLEVEL env var

daavoo · dberenbaum · commit 8a8025647b7e · 2023-08-17T13:03:33.000-04:00
diff --git a/src/dvclive/live.py b/src/dvclive/live.py
@@ -39,6 +39,7 @@
 )
 
 logger = logging.getLogger("dvclive")
+logger.setLevel(os.getenv(env.DVCLIVE_LOGLEVEL, "WARNING").upper())
 handler = logging.StreamHandler()
 formatter = logging.Formatter("%(levelname)s:%(name)s:%(message)s")
 handler.setFormatter(formatter)

Original file line number	Diff line number	Diff line change
`@@ -39,6 +39,7 @@`
`39`	`39`	`)`
`40`	`40`
`41`	`41`	`logger = logging.getLogger("dvclive")`
	`42`	`+logger.setLevel(os.getenv(env.DVCLIVE_LOGLEVEL, "WARNING").upper())`
`42`	`43`	`handler = logging.StreamHandler()`
`43`	`44`	`formatter = logging.Formatter("%(levelname)s:%(name)s:%(message)s")`
`44`	`45`	`handler.setFormatter(formatter)`