o `۷i@sddlmZddlmZddlZddlmZddlm Z ddl m Z ddl m Z mZddlmZdd lmZmZe Gd d d eZdS) ) FunctionType)DictN) InputReader) IOContext) JsonReader) DeveloperAPIoverride)SampleBatchType)registry_contains_inputregistry_get_inputc@sDeZdZdZedeeefdefddZ e e de fddZ d S) MixedInputaMixes input from a number of other input sources. .. testcode:: :skipif: True from ray.rllib.offline.io_context import IOContext from ray.rllib.offline.mixed_input import MixedInput ioctx = IOContext(...) MixedInput({ "sampler": 0.4, "/tmp/experiences/*.json": 0.4, "s3://bucket/expert.json": 0.2, }, ioctx) distioctxcCst|dkrtd|g|_g|_|D]D\}}|dkr*|j|n-t |t r8|j||nt |t rNt |rNt |}|j||n |jt|||j|qdS)zInitialize a MixedInput. Args: dist: dict mapping JSONReader paths or "sampler" to probabilities. The probabilities must sum to 1.0. ioctx: current IO context object. g?zValues must sum to 1.0: {}samplerN)sumvalues ValueErrorformatchoicespitemsappenddefault_sampler_input isinstancerstrr r r)selfr rkv input_creatorrS/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/rllib/offline/mixed_input.py__init__s  zMixedInput.__init__returncCstjj|j|jd}|S)N)r)nprandomchoicerrnext)rsourcerrr r&8szMixedInput.nextN)__name__ __module__ __qualname____doc__rrrfloatrr!rrr r&rrrr r s r )typesrtypingrnumpyr#ray.rllib.offline.input_readerrray.rllib.offline.io_contextrray.rllib.offline.json_readerrray.rllib.utils.annotationsrrray.rllib.utils.typingr ray.tune.registryr r r rrrr s