fix: use mock.patch on total_tqdm patch

This commit is contained in:
Dowon
2025-02-18 23:01:44 +09:00
parent 06100063b3
commit 941b7c7dfb

View File

@@ -49,12 +49,8 @@ def disable_safe_unpickle():
@contextmanager
def pause_total_tqdm():
orig = opts.data.get("multiple_tqdm", True)
try:
opts.data["multiple_tqdm"] = False
with patch.dict(opts.data, {"multiple_tqdm": False}, clear=False):
yield
finally:
opts.data["multiple_tqdm"] = orig
@contextmanager