Skip to content

Commit 7604800

Browse files
committed
Allow ToggleLogging to be reentrant
1 parent b3f6b6c commit 7604800

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

cmdstanpy/utils/logging.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,14 +31,16 @@ def get_logger() -> logging.Logger:
3131

3232
class ToggleLogging(AbstractContextManager):
3333
def __init__(self, disable: bool) -> None:
34+
self.disable = disable
3435
self.logger = get_logger()
3536
self.prev_state = self.logger.disabled
36-
self.logger.disabled = disable
37+
self.logger.disabled = self.disable
3738

3839
def __repr__(self) -> str:
3940
return ""
4041

4142
def __enter__(self) -> "ToggleLogging":
43+
self.logger.disabled = self.disable
4244
return self
4345

4446
def __exit__(

0 commit comments

Comments
 (0)