diff --git a/main_feasible.py b/main_feasible.py index 2a734df..0c7d256 100644 --- a/main_feasible.py +++ b/main_feasible.py @@ -32,3 +32,4 @@ plotting.plot_img_patch_w_outlines(img=par.oc.img, limits=par.world.limits, ax=ax) for qq in q: plotting.plot_x_path(x=qq, r=par.robot.spheres_rad, ax=ax, marker='o', alpha=0.5, color='k') +plotting.plt.show() \ No newline at end of file diff --git a/main_omp.py b/main_omp.py index 52df192..8d4f7c3 100644 --- a/main_omp.py +++ b/main_omp.py @@ -67,3 +67,4 @@ plotting.plot_x_path(x=q, r=par.robot.spheres_rad, ax=ax, marker='o', alpha=0.5) plotting.plot_x_path(x=q_opt, r=par.robot.spheres_rad, ax=ax, marker='o', color='k') +plotting.plt.show() \ No newline at end of file diff --git a/main_sample_start_end.py b/main_sample_start_end.py index d5b75d3..133c042 100644 --- a/main_sample_start_end.py +++ b/main_sample_start_end.py @@ -18,3 +18,4 @@ plotting.plot_circles(x=q[status == +1], r=radius, ax=ax, color='blue', alpha=0.5) plotting.plot_circles(x=q[status == -1], r=radius, ax=ax, color='red', alpha=0.5) +plotting.plt.show() \ No newline at end of file diff --git a/wzk/numpy2.py b/wzk/numpy2.py index 2f3356f..c1b23c0 100644 --- a/wzk/numpy2.py +++ b/wzk/numpy2.py @@ -267,7 +267,7 @@ def __fill_index_with(idx, axis, shape, mode='slice'): idx_with_ = [slice(None) for _ in range(len(shape)-len(axis))] elif mode == 'orange': - idx_with_ = np.ogrid[*[range(s) for i, s in enumerate(shape) if i not in axis]] + idx_with_ = np.ogrid(*[range(s) for i, s in enumerate(shape) if i not in axis]) else: raise ValueError(f"Unknown mode {mode}")