Skip to content

GitLab

Commit fb9f9ee6 authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 353699625
parent 32dcc1e4
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/dataloaders/dataset_fn.py
View file @ fb9f9ee6
...@@ -22,7 +22,7 @@ PossibleDatasetType = Union[Type[tf.data.Dataset], Callable[[tf.Tensor], Any]] ...@@ -22,7 +22,7 @@ PossibleDatasetType = Union[Type[tf.data.Dataset], Callable[[tf.Tensor], Any]]
def pick_dataset_fn(file_type: str) -> PossibleDatasetType: def pick_dataset_fn(file_type: str) -> PossibleDatasetType:
if file_type == 'tf_record': if file_type == 'tfrecord':
return tf.data.TFRecordDataset return tf.data.TFRecordDataset
raise ValueError('Unrecognized file_type: {}'.format(file_type)) raise ValueError('Unrecognized file_type: {}'.format(file_type))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment