Python remap_spec Examples

Example #1

File: t5.py Project: oceanfly/lit

 def output_spec(self):
     spec = lit_types.remap_spec(self.model.output_spec(),
                                 self.FIELD_RENAMES)
     spec["rougeL"] = lit_types.Scalar()
     return spec

Example #2

File: t5.py Project: oceanfly/lit

 def input_spec(self):
     spec = lit_types.remap_spec(self.model.input_spec(),
                                 self.FIELD_RENAMES)
     spec["source_language"] = lit_types.CategoryLabel()
     spec["target_language"] = lit_types.CategoryLabel()
     return spec

Example #3

File: t5.py Project: oceanfly/lit

 def input_spec(self):
     return lit_types.remap_spec(self.model.input_spec(),
                                 self.FIELD_RENAMES)

Example #4

File: t5.py Project: PAIR-code/lit

 def input_spec(self):
     return lit_types.remap_spec(self.wrapped.input_spec(),
                                 self.FIELD_RENAMES)