diff --git a/lib/annotate_rb/model_annotator/foreign_key_annotation/annotation.rb b/lib/annotate_rb/model_annotator/foreign_key_annotation/annotation.rb index cacb0c48..c9ef3a00 100644 --- a/lib/annotate_rb/model_annotator/foreign_key_annotation/annotation.rb +++ b/lib/annotate_rb/model_annotator/foreign_key_annotation/annotation.rb @@ -26,6 +26,10 @@ def to_markdown def to_default body.map(&:to_default).join("\n") end + + def to_yard + body.map(:&to_yard).join("\n") + end end end end diff --git a/lib/annotate_rb/model_annotator/index_annotation/annotation.rb b/lib/annotate_rb/model_annotator/index_annotation/annotation.rb index c735782b..24fecfec 100644 --- a/lib/annotate_rb/model_annotator/index_annotation/annotation.rb +++ b/lib/annotate_rb/model_annotator/index_annotation/annotation.rb @@ -26,6 +26,10 @@ def to_markdown def to_default body.map(&:to_default).join("\n") end + + def to_yard + body.map(:&to_yard).join("\n") + end end end end