def environment_variables()

in submitit/core/utils.py [0:0]

• 9 lines of code
• 4 McCabe index (conditional complexity)

def environment_variables(**kwargs: str) -> Iterator[None]:
    backup = {x: os.environ[x] for x in kwargs if x in os.environ}
    os.environ.update(kwargs)
    try:
        yield
    finally:
        for x in kwargs:
            del os.environ[x]
        os.environ.update(backup)