Python _get_dataset_files Exemples

Exemple #1

 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})

Exemple #2

 def _get_files(self, instruction):
     return tfrecords_reader._get_dataset_files('mnist', '/foo/bar',
                                                instruction,
                                                self.NAME2SHARD_LENGTHS)