J is part of graph, not param.

Author: SebKrantz

src/models/model_fixed.jl

@@ -35,7 +35,7 @@ function model_fixed(optimizer, auxdata)
     # Define Yjn (production) as expression
     @expression(model, Yjn[j=1:graph.J, n=1:param.N], graph.Zjn[j, n] * Ljn[j, n]^param.a)
     # Balanced flow constraints
-    @constraint(model, Pjn[j in 1:param.J, n in 1:param.N],
+    @constraint(model, Pjn[j in 1:graph.J, n in 1:param.N],
         Cjn[j, n] + sum(A[j, i] * Qin[i, n] for i in 1:graph.ndeg) -
         Yjn[j, n] + sum(
             ifelse(Qin[i, n] > 0, Apos[j, i], Aneg[j, i]) *

src/models/model_fixed_armington.jl

@@ -34,7 +34,7 @@ function model_fixed_armington(optimizer, auxdata)
     @objective(model, Max, U)
 
     # Balanced flow constraints
-    @constraint(model, Pjn[j in 1:param.J, n in 1:param.N],
+    @constraint(model, Pjn[j in 1:graph.J, n in 1:param.N],
         Cjn[j, n] + sum(A[j, i] * Qin[i, n] for i in 1:graph.ndeg) -
         Yjn[j, n] + sum(
             ifelse(Qin[i, n] > 0, Apos[j, i], Aneg[j, i]) *

src/models/model_mobility.jl

@@ -39,7 +39,7 @@ function model_mobility(optimizer, auxdata)
     # balanced flow constraints
     # Yjn = @expression(model, graph.Zjn .* Ljn .^ param.a) # Same thing
     @expression(model, Yjn[j=1:graph.J, n=1:param.N], graph.Zjn[j, n] * Ljn[j, n]^param.a)
-    @constraint(model, Pjn[j in 1:param.J, n in 1:param.N],
+    @constraint(model, Pjn[j in 1:graph.J, n in 1:param.N],
         Cjn[j, n] + sum(A[j, i] * Qin[i, n] for i in 1:graph.ndeg) -
         Yjn[j, n] + sum(
             ifelse(Qin[i, n] > 0, Apos[j, i], Aneg[j, i]) *

src/models/model_mobility_armington.jl

@@ -37,7 +37,7 @@ function model_mobility_armington(optimizer, auxdata)
 
     # balanced flow constraints
      @expression(model, Yjn[j=1:graph.J, n=1:param.N], graph.Zjn[j, n] * Lj[j]^param.a)
-    @constraint(model, Pjn[j in 1:param.J, n in 1:param.N],
+    @constraint(model, Pjn[j in 1:graph.J, n in 1:param.N],
         Cjn[j, n] + sum(A[j, i] * Qin[i, n] for i in 1:graph.ndeg) -
         Yjn[j, n] + sum(
             ifelse(Qin[i, n] > 0, Apos[j, i], Aneg[j, i]) *

src/models/model_partial_mobility.jl

@@ -57,7 +57,7 @@ function model_partial_mobility(optimizer, auxdata)
     # Balanced flow constraints: same as with unrestricted mobility (no restrictions on goods)
     # Yjn = @expression(model, graph.Zjn .* Ljn .^ param.a) # Same thing
     @expression(model, Yjn[j=1:graph.J, n=1:param.N], graph.Zjn[j, n] * Ljn[j, n]^param.a)
-    @constraint(model, Pjn[j in 1:param.J, n in 1:param.N],
+    @constraint(model, Pjn[j in 1:graph.J, n in 1:param.N],
         Cjn[j, n] + sum(A[j, i] * Qin[i, n] for i in 1:graph.ndeg) -
         Yjn[j, n] + sum(
             ifelse(Qin[i, n] > 0, Apos[j, i], Aneg[j, i]) *

src/models/model_partial_mobility_armington.jl

@@ -53,7 +53,7 @@ function model_partial_mobility_armington(optimizer, auxdata)
 
     # Balanced flow constraints: same as with unrestricted mobility (no restrictions on goods)
     @expression(model, Yjn[j=1:graph.J, n=1:param.N], graph.Zjn[j, n] * Lj[j]^param.a)
-    @constraint(model, Pjn[j in 1:param.J, n in 1:param.N],
+    @constraint(model, Pjn[j in 1:graph.J, n in 1:param.N],
         Cjn[j, n] + sum(A[j, i] * Qin[i, n] for i in 1:graph.ndeg) -
         Yjn[j, n] + sum(
             ifelse(Qin[i, n] > 0, Apos[j, i], Aneg[j, i]) *