def testConvertFieldsForSpec(self): fields = [ extension_type_field.ExtensionTypeField('x', int), extension_type_field.ExtensionTypeField( 'y', typing.Tuple[typing.Union[int, bool], ...]), extension_type_field.ExtensionTypeField('z', ops.Tensor) ] field_values = { 'x': 1, 'y': [1, True, 3], 'z': tensor_spec.TensorSpec([5, 3]) } extension_type_field.convert_fields_for_spec(fields, field_values) self.assertEqual(set(field_values), set(['x', 'y', 'z'])) self.assertEqual(field_values['x'], 1) self.assertEqual(field_values['y'], (1, True, 3)) self.assertEqual(field_values['z'], tensor_spec.TensorSpec([5, 3]))
def _tf_extension_type_convert_fields(self): extension_type_field.convert_fields_for_spec( self._tf_extension_type_fields(), self.__dict__)