lib/swagalicious/swagger_formatter.rb in swagalicious-0.5.2 vs lib/swagalicious/swagger_formatter.rb in swagalicious-0.5.3

@@ -53,10 +53,10 @@
     end
 
     private
 
     def pretty_generate(doc)
-      if config.swagger_format == :yaml
+      if config.swagger_format == :yaml || doc.delete(:swagger_format) == :yaml
         clean_doc = yaml_prepare(doc)
         YAML.dump(clean_doc)
       else # config errors are thrown in "def swagger_format", no throw needed here
         JSON.pretty_generate(doc)
       end