lib/generators/rom/relation/templates/relation.rb.erb in rom-rails-0.3.1 vs lib/generators/rom/relation/templates/relation.rb.erb in rom-rails-0.3.2

@@ -1,6 +1,14 @@
 class <%= class_name %>Relation < ROM::Relation<%= "[:#{adapter}]" %>
+<% if repository -%>
+  repository :<%= repository %>
+<% else -%>
+  # repository :default
+<% end -%>
+
   dataset :<%= dataset %>
+
+  register_as :<%= register_as %>
 
   # define your methods here ie:
   #
   # def all
   #   select(:id, :name).order(:id)