Override
  public boolean isRainbowType(TextAttributesKey type) {
    return PyRainbowVisitor.Holder.getHIGHLIGHTING_KEYS().contains(type);
  }

Override
  public boolean isRainbowType(TextAttributesKey type) 