docs: def _clean_annotations(annotations_dict: AnnotationsHint) -> AnnotationsDict:
      """Fix the formatting of annotation dict."""
      return {
          key: (
              values if isinstance(values, dict) else
              {v: True for v in values} if isinstance(values, set) else
              {values: True}
          )
          for key, values in annotations_dict.items()
      }