def swift_default()

in Sources/TensorFlow/Bindings/generate_wrappers.py [0:0]

• 32 lines of code
• 17 McCabe index (conditional complexity)

  def swift_default(self):
    def swift_float(f):
      if f == float('inf'): return 'Double.infinity'
      if f == float('-inf'): return '-Double.infinity'
      return '%g' % f

    if not self.is_inferred_type_attr and self.attr_def.default_value:
      default_value = self.attr_def.default_value
      if default_value.HasField('b'):
        default_value = str(default_value.b).lower()
      elif default_value.HasField('i'):
        default_value = str(default_value.i)
      elif default_value.HasField('f'):
        default_value = swift_float(default_value.f)
      elif default_value.HasField('s') and default_value.s:
        s = str(default_value.s)
        default_value = '.' + swift_compatible_identifier(s.lower()) \
            if self._use_enum else json.dumps(s) # '"' + s + '"'
      elif default_value.HasField('list'):
        if default_value.list.i:
          default_values = [str(s) for s in default_value.list.i]
          default_value = '[' + ', '.join(default_values) + ']'
        elif default_value.list.f:
          default_values = [swift_float(s) for s in default_value.list.f]
          default_value = '[' + ', '.join(default_values) + ']'
        else:
          default_value = None
      else:
        default_value = None
      if default_value is not None:
        default_value = default_value.replace("\t", "\\t")
        return ' = ' + default_value
    return ''