Update modelling_magiv2.py

modelling_magiv2.py

@@ -146,7 +146,7 @@ class Magiv2Model(PreTrainedModel):
         problem.solve()
         assignments = []
         for v in problem.variables():
-            if v.varValue > 0:
+            if v.varValue is not None and v.varValue > 0:
                 index, assignment = v.name.split("(")[1].split(")")[0].split(",")
                 assignment = assignment[1:]
                 assignments.append((int(index), int(assignment)))