def expand(self, pvalue):
     return pvalue.pipeline | beam.io.ReadFromText(
         self._path, coder=mlcoders.TrainingJobResultCoder())
 def expand(self, result):
     return (result
             | beam.io.textio.WriteToText(
                 self._path,
                 shard_name_template='',
                 coder=mlcoders.TrainingJobResultCoder()))