Source code for ml.utils.random

"""Function(s) for dealing with random numbers."""

import random

import numpy as np
import torch

from ml.core.env import get_env_random_seed


def set_random_seed(seed: int | None = None, offset: int = 0) -> None:
    if seed is None:
        seed = get_env_random_seed()
    seed += offset
    random.seed(seed)
    np.random.seed(seed)
    torch.manual_seed(seed)