lib/hillary/repo/version.rb in hillary-0.0.3 vs lib/hillary/repo/version.rb in hillary-0.0.4

@@ -2,11 +2,11 @@
 
 module Hillary
   class Repo
     class Version
       BRANCHES = [
-        DEV = 'dev',
-        MASTER = 'master'
+        DEV = ENV['DEV_BRANCH'] || 'dev',
+        MASTER = ENV['MASTER_BRANCH'] || 'master'
       ]
 
       InvalidRepositoryState = Class.new(StandardError)
 
       class << self