github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-01-15 04:47:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

tidy: remove unused utils

Browse Source
This commit is contained in:
psychedelicious
2025-03-05 10:31:11 +10:00
parent c1cf01a038
commit d2f8db9745
1 changed files with 0 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
tests/env_var_utils.py
View File

@@ -1,25 +0,0 @@
import os
from contextlib import contextmanager
@contextmanager
def unset_env_var(env_var: str):
"""Context manager that unsets an environment variable, re-setting it to the original value when the context is exited."""
prev_value = os.environ.get(env_var, None)
if prev_value is not None:
del os.environ[env_var]
yield
if prev_value is not None:
os.environ[env_var] = prev_value
@contextmanager
def set_env_var(env_var: str, value: str):
"""Context manager that sets an environment variable, re-setting it to the original value when the context is exited."""
prev_value = os.environ.get(env_var, None)
os.environ[env_var] = value
yield
if prev_value is not None:
os.environ[env_var] = prev_value
else:
del os.environ[env_var]