Example of managing NumPy seed with contextmanager

np_seed_contextmanager.py

import contextlib
import numpy as np

# For info on contextlib, see https://docs.python.org/3/library/contextlib.html
@contextlib.contextmanager
def local_seed(seed):
    state = np.random.get_state()
    np.random.seed(seed)
    try:
        yield
    finally:
        np.random.set_state(state)


if __name__ == '__main__':
    my_seed = 42
    with local_seed(my_seed):
        value = np.random.rand(1)
        print(f"sampled {value} using seed {my_seed}")