lib/rails_erd/domain/entity.rb in rails-erd-1.3.0 vs lib/rails_erd/domain/entity.rb in rails-erd-1.3.1

@@ -38,10 +38,10 @@
         @domain, @name, @model = domain, name, model
       end
 
       # Returns an array of attributes for this entity.
       def attributes
-        @attributes ||= if generalized? then [] else Attribute.from_model(domain, model) end
+        @attributes ||= generalized? ? [] : Attribute.from_model(domain, model)
       end
 
       # Returns an array of all relationships that this entity has with other
       # entities in the domain model.
       def relationships