CombinatorialOptimisation.train() runs but gives wrong answers!