Get a buffer reader for the given dataset name.
(config: BufferStorageConfig)
| 12 | |
| 13 | |
| 14 | def get_buffer_reader(config: BufferStorageConfig) -> BufferReader: |
| 15 | """Get a buffer reader for the given dataset name.""" |
| 16 | if not isinstance(config, StorageConfig): |
| 17 | storage_config: StorageConfig = config.to_storage_config() |
| 18 | else: |
| 19 | storage_config = config |
| 20 | reader_cls = READER.get(storage_config.storage_type) |
| 21 | if reader_cls is None: |
| 22 | raise ValueError(f"{storage_config.storage_type} not supported.") |
| 23 | return reader_cls(storage_config) |
| 24 | |
| 25 | |
| 26 | def get_buffer_writer(config: BufferStorageConfig) -> BufferWriter: |