Skip to content

Commit 4e02d79

Browse files
add logging for expectation function
1 parent 1f58cb4 commit 4e02d79

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

varipeps/config.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -325,6 +325,7 @@ class VariPEPS_Config:
325325
log_level_optimizer: LogLevel = LogLevel.INFO
326326
log_level_ctmrg: LogLevel = LogLevel.INFO
327327
log_level_line_search: LogLevel = LogLevel.INFO
328+
log_level_expectation: LogLevel = LogLevel.INFO
328329
log_to_console: bool = True
329330
log_to_file: bool = False
330331
log_file: str = "varipeps.log"

varipeps/utils/logging_config.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,9 @@ def init_logging(cfg: Any | None = None) -> None:
6464
logging.getLogger("varipeps.line_search").setLevel(
6565
_to_py_log_level(getattr(cfg, "log_level_line_search", logging.INFO))
6666
)
67+
logging.getLogger("varipeps.expectation").setLevel(
68+
_to_py_log_level(getattr(cfg, "log_level_expectation", logging.INFO))
69+
)
6770

6871
_LOGGING_INITIALIZED = True
6972

0 commit comments

Comments
 (0)