tests/__init__.py

from constants import PROJECT_ROOT_FOLDER


TEST_ROOT_FOLDER = PROJECT_ROOT_FOLDER / "tests"
EVENTS_FOLDER = TEST_ROOT_FOLDER / "events"


def set_seed(seed=0):
    import random
    import torch
    import numpy as np

    random.seed(seed)
    np.random.seed(seed)
    torch.manual_seed(seed)