Commit fd5849c9 authored by YaningGao's avatar YaningGao
Browse files

Merge branch 'main' of github.com:RAGEN-AI/vagen

parents 598818c3 08291d44
Loading
Loading
Loading
Loading
+7 −1
Original line number Diff line number Diff line
from dataclasses import dataclass, field
from abc import ABC, abstractmethod
from typing import Optional, List, Union
import random
@dataclass
class BaseEnvConfig(ABC):
    format_reward: float = 0.5
@@ -19,3 +20,8 @@ class BaseEnvConfig(ABC):
            default: Default value if key is not found
        """
        return getattr(self, key, default)
    
    def generate_seeds(self,size,seed=0,n_candidate: int = 20000,) -> list:
        random.seed(seed)
        seeds=random.sample(range(0, n_candidate+size), size)
        return seeds
 No newline at end of file
+20 −0
Original line number Diff line number Diff line
@@ -9,10 +9,30 @@ from contextlib import contextmanager
import os

def permanent_seed(seed: int) -> None:
    """Set all random seeds for reproducibility across multiple libraries.
    
    Args:
        seed: Integer seed value to use across all random number generators
    """
    import random
    import numpy as np
    import torch
    import os
    
    # Basic Python random
    random.seed(seed)
    
    # NumPy
    np.random.seed(seed)
    
    # PyTorch
    torch.manual_seed(seed)
    torch.backends.cudnn.deterministic = True
    torch.backends.cudnn.benchmark = False
    
    # CUDA
    if torch.cuda.is_available():
        torch.cuda.manual_seed(seed)
        torch.cuda.manual_seed_all(seed)