mirror of
https://github.com/vale981/ray
synced 2025-03-06 02:21:39 -05:00
18 lines
611 B
Python
18 lines
611 B
Python
from ray.rllib.offline.dataset_reader import (
|
|
_get_resource_bundles as dataset_reader_get_resource_bundles,
|
|
)
|
|
from ray.rllib.utils.annotations import PublicAPI
|
|
from ray.rllib.utils.typing import PartialTrainerConfigDict
|
|
from typing import Dict, List
|
|
|
|
|
|
@PublicAPI
|
|
def get_offline_io_resource_bundles(
|
|
config: PartialTrainerConfigDict,
|
|
) -> List[Dict[str, float]]:
|
|
# DatasetReader is the only offline I/O component today that
|
|
# requires compute resources.
|
|
if config["input"] == "dataset":
|
|
return dataset_reader_get_resource_bundles(config["input_config"])
|
|
else:
|
|
return []
|