Gymnasium/tests/wrappers/monitoring/helpers.py

mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-19 13:32:03 +00:00
import contextlib
import shutil
import tempfile


@contextlib.contextmanager
def tempdir():
    temp = tempfile.mkdtemp()
    yield temp
    shutil.rmtree(temp)
Discard viewer object after render with close=True Fixes #95 2016-05-15 17:22:38 -07:00			`import contextlib`
			`import shutil`
			`import tempfile`

Blacken the codebase (#2265) 2021-07-29 02:26:34 +02:00
Discard viewer object after render with close=True Fixes #95 2016-05-15 17:22:38 -07:00			`@contextlib.contextmanager`
			`def tempdir():`
			`temp = tempfile.mkdtemp()`
			`yield temp`
			`shutil.rmtree(temp)`