Improved code [skip ci]

ankane · Oct 6, 2024 · dfc2cac · dfc2cac
1 parent f75fc13
commit dfc2cac
Showing 1 changed file with 11 additions and 15 deletions.
diff --git a/lib/or_tools/solver.rb b/lib/or_tools/solver.rb
@@ -7,21 +7,17 @@ def sum(arr)
     def add(expr)
       left, op, const = Utils.index_constraint(expr)
 
-      case op
-      when :<=
-        lb = -infinity
-        ub = const
-      when :>=
-        lb = const
-        ub = infinity
-      when :==
-        lb = const
-        ub = const
-      else
-        raise "todo: #{op}"
-      end
-
-      constraint = constraint(lb, ub)
+      constraint =
+        case op
+        when :<=
+          self.constraint(-infinity, const)
+        when :>=
+          self.constraint(const, infinity)
+        when :==
+          self.constraint(const, const)
+        else
+          raise "todo: #{op}"
+        end
       left.each do |var, c|
         constraint.set_coefficient(var, c)
       end