Fix format

Author: blegat

src/RelativeEntropy/RelativeEntropy.jl

@@ -55,17 +55,28 @@ function _exponents_matrix(monos)
 end
 
 struct SignomialSAGESet <: PolyJuMP.PolynomialSet end
-JuMP.reshape_set(::SignomialSAGECone, ::PolyJuMP.PolynomialShape) = SignomialSAGESet()
-JuMP.moi_set(::SignomialSAGESet, monos) = SignomialSAGECone(_exponents_matrix(monos))
+function JuMP.reshape_set(::SignomialSAGECone, ::PolyJuMP.PolynomialShape)
+    return SignomialSAGESet()
+end
+function JuMP.moi_set(::SignomialSAGESet, monos)
+    return SignomialSAGECone(_exponents_matrix(monos))
+end
 
 struct PolynomialSAGESet <: PolyJuMP.PolynomialSet end
-JuMP.reshape_set(::PolynomialSAGECone, ::PolyJuMP.PolynomialShape) = PolynomialSAGESet()
-JuMP.moi_set(::PolynomialSAGESet, monos) = PolynomialSAGECone(_exponents_matrix(monos))
+function JuMP.reshape_set(::PolynomialSAGECone, ::PolyJuMP.PolynomialShape)
+    return PolynomialSAGESet()
+end
+function JuMP.moi_set(::PolynomialSAGESet, monos)
+    return PolynomialSAGECone(_exponents_matrix(monos))
+end
 
 struct SignomialAGESet{MT<:MP.AbstractMonomial} <: PolyJuMP.PolynomialSet
     monomial::MT
 end
-function JuMP.reshape_set(set::SignomialAGECone, shape::PolyJuMP.PolynomialShape)
+function JuMP.reshape_set(
+    set::SignomialAGECone,
+    shape::PolyJuMP.PolynomialShape,
+)
     return SignomialAGESet(shape.monomials[set.k])
 end
 function JuMP.moi_set(set::SignomialAGESet, monos)
@@ -76,7 +87,10 @@ end
 struct PolynomialAGESet{MT<:MP.AbstractMonomial} <: PolyJuMP.PolynomialSet
     monomial::MT
 end
-function JuMP.reshape_set(set::PolynomialAGECone, shape::PolyJuMP.PolynomialShape)
+function JuMP.reshape_set(
+    set::PolynomialAGECone,
+    shape::PolyJuMP.PolynomialShape,
+)
     return PolynomialAGESet(shape.monomials[set.k])
 end
 function JuMP.moi_set(set::PolynomialAGESet, monos)
@@ -138,5 +152,4 @@ function PolyJuMP.bridges(
     return [(SignomialBridge, PolyJuMP._coef_type(F))]
 end
 
-
 end

src/RelativeEntropy/bridges/sage.jl

@@ -1,6 +1,8 @@
 struct SAGEBridge{T,F,G} <: MOI.Bridges.Constraint.AbstractBridge
     ν::Matrix{MOI.VariableIndex}
-    age_constraints::Vector{MOI.ConstraintIndex{MOI.VectorOfVariables,SignomialAGECone}}
+    age_constraints::Vector{
+        MOI.ConstraintIndex{MOI.VectorOfVariables,SignomialAGECone},
+    }
     equality_constraints::Vector{MOI.ConstraintIndex{F,MOI.EqualTo{T}}}
 end
 
@@ -13,7 +15,10 @@ function MOI.Bridges.Constraint.bridge_constraint(
     m = size(set.α, 1)
     ν = Matrix{MOI.VariableIndex}(undef, m, m)
     age_constraints =
-        Vector{MOI.ConstraintIndex{MOI.VectorOfVariables,SignomialAGECone}}(undef, m)
+        Vector{MOI.ConstraintIndex{MOI.VectorOfVariables,SignomialAGECone}}(
+            undef,
+            m,
+        )
     for k in 1:m
         ν[k, :], age_constraints[k] =
             MOI.add_constrained_variables(model, SignomialAGECone(set.α, k))

test/relative_entropy.jl

@@ -22,8 +22,20 @@ function _test_motzkin(x, y, T, solver, set)
 end
 
 function test_motzkin(x, y, T, solver)
-    _test_motzkin(x, y, T, solver, PolyJuMP.RelativeEntropy.SignomialAGESet(x^2 * y^2))
-    _test_motzkin(x, y, T, solver, PolyJuMP.RelativeEntropy.SignomialSAGESet())
+    _test_motzkin(
+        x,
+        y,
+        T,
+        solver,
+        PolyJuMP.RelativeEntropy.SignomialAGESet(x^2 * y^2),
+    )
+    return _test_motzkin(
+        x,
+        y,
+        T,
+        solver,
+        PolyJuMP.RelativeEntropy.SignomialSAGESet(),
+    )
 end
 
 import ECOS