j^ddlZddlmZmZddlmZddlmZGddejZ y)N)ListOptional) dataset_item)base_dataset_samplerc reZdZ d dededeeddfdZdeejdeejfdZ y) RandomDatasetSamplerN max_samplesshuffleseedreturnc.||_||_||_y)aSamples a random subset of dataset items. This class is a dataset sampler that selects a random subset of items from a dataset. The number of items to sample can be specified, and shuffling can be enabled or disabled. An optional random seed can be provided for reproducibility. Args: max_samples: The maximum number of samples to generate. shuffle: Whether to shuffle the samples. Default is True, False provides a speedup for large datasets. seed: Seed for the random number generator. If None, then fresh, unpredictable entropy will be pulled from the OS. N)r r r )selfr r r s /Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/evaluation/samplers/random_dataset_sampler.py__init__zRandomDatasetSampler.__init__ s '  data_itemsct|dk(rgStj|j}t t||j }|j ||}|jr|j||S)Nr)lenrandomRandomr minr sampler )rrrng sample_sizeitemss rrzRandomDatasetSampler.samplesk z?a ImmDII&#j/4+;+;<  :{3 << KK  r)TN) __name__ __module__ __qualname__intboolrrrr DatasetItemrrrr r sYLP)-r(s$ !1"'/BB'r