code: def _toolkits_select_columns(dataset, columns):
      
      try:
          return dataset.select_columns(columns)
      except RuntimeError:
          missing_features = list(set(columns).difference(set(dataset.column_names())))
          raise ToolkitError("Input data does not contain the following columns: "  +
                  "{}".format(missing_features))
docs: Same as select columns but redirect runtime error to ToolkitError.