| 774 | expected_output=[ragged_factory_ops.constant([[i]]) for i in range(5)]) |
| 775 | |
| 776 | def testRaggedChain(self): |
| 777 | |
| 778 | def _ragged(i): |
| 779 | return ragged_tensor.RaggedTensor.from_tensor(i * [[1]]) |
| 780 | |
| 781 | def _concat(i): |
| 782 | self.assertTrue(ragged_tensor.is_ragged(i)) |
| 783 | return ragged_concat_ops.concat([i, i], 0) |
| 784 | |
| 785 | dataset = dataset_ops.Dataset.range(10).map(_ragged).map(_concat) |
| 786 | |
| 787 | self.assertDatasetProduces( |
| 788 | dataset, |
| 789 | expected_output=[ |
| 790 | self.evaluate(_concat(ragged_factory_ops.constant([[i]]))) |
| 791 | for i in range(10) |
| 792 | ]) |
| 793 | |
| 794 | @test_util.run_v1_only("b/123904513") |
| 795 | def testParallelMapOutOfRangeError(self): |