def test_missing_shard_lengths(self): instruction = tfrecords_reader._AbsoluteInstruction( 'train', None, None) with self.assertRaisesWithPredicateMatch( AssertionError, 'S3 tfrecords_reader cannot be used'): tfrecords_reader._get_dataset_files('mnist', '/foo/bar', instruction, {'train': None})
def _get_files(self, instruction): return tfrecords_reader._get_dataset_files('mnist', '/foo/bar', instruction, self.NAME2SHARD_LENGTHS)