Skip to content

Commit

Permalink
int problem
Browse files Browse the repository at this point in the history
  • Loading branch information
rvandaele committed Mar 23, 2019
1 parent 415db2d commit 3ebd04b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions run.py
Original file line number Diff line number Diff line change
Expand Up @@ -155,7 +155,7 @@ def compute_final_solution_phase_3(xc, yc, probability_map_phase_2, ncandidates,
g.append('f2_%d_%d' % (ip, ipl), nodes[ipl])

for i in range(nldms):
v = probability_map_phase_2[np.array(y_candidates[i]), np.array(x_candidates[i]), i]
v = probability_map_phase_2[np.array(y_candidates[i]).astype('int'), np.array(x_candidates[i]).astype('int'), i]
g.add(Factor('f1_%d' % i, v / np.sum(v)))
g.append('f1_%d' % i, nodes[i])

Expand Down Expand Up @@ -258,4 +258,4 @@ def main():
conn.job.update(status=Job.TERMINATED, progress=100, statusComment="Finished.")

if __name__ == "__main__":
main()
main()

0 comments on commit 3ebd04b

Please sign in to comment.