def print_node_type_info()

in src/neuron-gatherinfo/mx_neuron_check_model.py [0:0]

• 34 lines of code
• 10 McCabe index (conditional complexity)

  def print_node_type_info(self):
    self.cnt_total = len(self.nodetypes)
    self.cnt_supported = 0
    if self.parser_args.show_names:
      widthn = max(max(map(len, self.nodenames)), 8)
      widtht = max(max(map(len, self.nodetypes)), 8)
      format_str = "{:<" + str(widthn) + "}  {:<" + str(widtht) + "}  {:<4}"
      pp = lambda x: print(format_str.format(*x))
      pp(['Op Name', 'Op Type', 'Neuron Supported ?'])
      pp(['-------', '-------', '------------------'])
      for idx, opname in enumerate(self.nodenames):
        optype = self.nodetypes[idx]
        if optype in self.supported:
          pp([opname, optype, 'Yes'])
          self.cnt_supported += 1
      for idx, opname in enumerate(self.nodenames):
        optype = self.nodetypes[idx]
        if optype not in self.supported:
          pp([opname, optype, 'No'])
    else:
      count = Counter(self.nodetypes)
      width = max(max(map(len, self.nodetypes)), 8)
      format_str = "{:<" + str(width) + "}  {:<14}  {:<4}"
      pp = lambda x: print(format_str.format(*x))
      pp(['Op Type', 'Num Instances', 'Neuron Supported ?'])
      pp(['-------', '-------------', '------------------'])
      for key in count:
        if key in self.supported:
          pp([key, count[key], 'Yes'])
          self.cnt_supported += count[key]
      for key in count:
        if key not in self.supported:
          pp([key, count[key], 'No'])
    print()