Skip to content

Commit

Permalink
Internal change
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 584305365
  • Loading branch information
tomvdw authored and SeqIO committed Nov 21, 2023
1 parent 27bdb22 commit 64c56b8
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
3 changes: 2 additions & 1 deletion seqio/dataset_providers.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
Defines Tasks, TaskRegistry, Mixture, and MixtureRegistry
"""

from __future__ import annotations

import abc
Expand Down Expand Up @@ -499,7 +500,7 @@ def __init__(
caching_permitted: indicates whether this data source may be cached.
Default True.
decoders: dict (optional), mapping from features to tfds.decode.Decoders,
such as tfds.decode.SkipDecoding() for skipping image byte decoding
such as tfds.decode.SkipDecoding() for skipping image byte decoding.
"""
if splits and not isinstance(splits, dict):
splits = {k: k for k in splits}
Expand Down
2 changes: 1 addition & 1 deletion seqio/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -135,7 +135,7 @@ def __init__(
'train[':1%']), or `TfdsSplit` (e.g. `TfdsSplit(dataset='mnist',
split='train')`). If `TfdsSplit` are used then `name` must be empty.
decoders: dict (optional), mapping from features to tfds.decode.Decoders,
such as tfds.decode.SkipDecoding() for skipping image byte decoding
such as tfds.decode.SkipDecoding() for skipping image byte decoding.
"""
_validate_tfds_name(name)
self._name = name
Expand Down

0 comments on commit 64c56b8

Please sign in to comment.