utils.py

#  utils.py

import random
import torch
from config import MAX_SEED # Import MAX_SEED from config

def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
    if randomize_seed:
        seed = random.randint(0, MAX_SEED)
    return seed