Ensures that the ``dataloader`` is a :class:`.DataSpec`. Args: dataloader (DataSpec | Iterable | dict): A DataSpec, DataLoader, or dict of DataSpec kwargs. Returns: DataSpec: A DataSpec
(dataloader: Union[DataSpec, Iterable, dict])
| 319 | |
| 320 | |
| 321 | def ensure_data_spec(dataloader: Union[DataSpec, Iterable, dict]) -> DataSpec: |
| 322 | """Ensures that the ``dataloader`` is a :class:`.DataSpec`. |
| 323 | |
| 324 | Args: |
| 325 | dataloader (DataSpec | Iterable | dict): A DataSpec, DataLoader, or dict of DataSpec kwargs. |
| 326 | |
| 327 | Returns: |
| 328 | DataSpec: A DataSpec |
| 329 | """ |
| 330 | if isinstance(dataloader, dict): |
| 331 | # treat as kwargs for DataSpec |
| 332 | dataloader = DataSpec(**dataloader) |
| 333 | if not isinstance(dataloader, DataSpec): |
| 334 | dataloader = DataSpec(dataloader) |
| 335 | |
| 336 | return dataloader |