Add hack to use erd patch only for RAILS_ENV=development only

1 files changed, 10 insertions, 8 deletions

diff --git a/config/initializers/relationship.rb b/config/initializers/relationship.rb
index 6f3fc29be..492aa627f 100644
--- a/config/initializers/relationship.rb
+++ b/config/initializers/relationship.rb
@@ -1,13 +1,15 @@
-require 'rails_erd/domain/relationship'
+if Rails.env.development?
+  require 'rails_erd/domain/relationship'
 
-module RailsERD
-  class Domain
-    class Relationship
-      class << self
-        private
+  module RailsERD
+    class Domain
+      class Relationship
+        class << self
+          private
 
-        def association_identity(association)
-          Set[association_owner(association), association_target(association)]
+          def association_identity(association)
+            Set[association_owner(association), association_target(association)]
+          end
         end
       end
     end