lib/rails_erd/tasks.rake in rails-erd-1.5.2 vs lib/rails_erd/tasks.rake in rails-erd-1.6.0

@@ -17,10 +17,16 @@
     (RailsERD.options.keys.map(&:to_s) & ENV.keys).each do |option|
       RailsERD.options[option.to_sym] = case ENV[option]
       when "true", "yes" then true
       when "false", "no" then false
       when /,/ then ENV[option].split(/\s*,\s*/)
-      else ENV[option].to_sym
+      when /^\d+$/ then ENV[option].to_i
+      else
+        if option == 'only'
+          [ENV[option]]
+        else
+          ENV[option].to_sym
+        end
       end
     end
   end
 
   task :load_models do