lib/rails_erd/domain.rb in rails-erd-1.0.0 vs lib/rails_erd/domain.rb in rails-erd-1.1.0

@@ -112,18 +112,18 @@
         end
       end
     end
 
     def models
-      @models ||= @source_models.reject(&:abstract_class?).select { |model| check_model_validity(model) }
+      @models ||= @source_models.select { |model| check_model_validity(model) }
     end
 
     def associations
       @associations ||= models.collect(&:reflect_on_all_associations).flatten.select { |assoc| check_association_validity(assoc) }
     end
 
     def check_model_validity(model)
-      model.table_exists? or raise "table #{model.table_name} does not exist"
+      model.abstract_class? or model.table_exists? or raise "table #{model.table_name} does not exist"
     rescue => e
       warn "Ignoring invalid model #{model.name} (#{e.message})"
     end
 
     def check_association_validity(association)