diff --git a/lib/generators/scenic/model/model_generator.rb b/lib/generators/scenic/model/model_generator.rb index b04e4186..41627fcc 100644 --- a/lib/generators/scenic/model/model_generator.rb +++ b/lib/generators/scenic/model/model_generator.rb @@ -36,12 +36,23 @@ def invoke_view_generator def evaluate_template(source) source = File.expand_path(find_in_source_paths(source.to_s)) context = instance_eval("binding", __FILE__, __LINE__) - ERB.new( - ::File.binread(source), - nil, - "-", - "@output_buffer", - ).result(context) + + if ERB.instance_method(:initialize).parameters.assoc(:key) # Ruby 2.6+ + erb = ERB.new( + ::File.binread(source), + trim_mode: "-", + eoutvar: "@output_buffer", + ) + else + erb = ERB.new( + ::File.binread(source), + nil, + "-", + "@output_buffer", + ) + end + + erb.result(context) end def generating?