Merge pull request #357 from astro-informatics/tk/debug_examples

Fix bugs in examples
astro-informatics · Oct 9, 2024 · 4ad17e3 · 4ad17e3
2 parents 2537a2b + 98335ab
commit 4ad17e3
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/cpp/example/mem_w_algos.cc b/cpp/example/mem_w_algos.cc
@@ -58,7 +58,7 @@ int main(int nargs, char const **args) {
   std::vector<t_int> image_index;
   std::tie(uv_data, image_index, w_stacks) = utilities::w_stacking_with_all_to_all(
       uv_data, du, Ju, Jw, world, 100, 0., cost, k_means_rel_diff);
-  for (t_real k = 0; k < uv_data.size(); k++) {
+  for (t_int k = 0; k < uv_data.size(); k++) {
     const t_int Ju_max = widefield::w_support(uv_data.w(k) - w_stacks.at(image_index.at(k)), du,
                                               static_cast<t_int>(Ju), static_cast<t_int>(Jw));
     const t_real mem = (Ju_max * Ju_max) * 16.;

diff --git a/cpp/example/padmm_real_data.cc b/cpp/example/padmm_real_data.cc
@@ -46,7 +46,7 @@ void padmm(const std::string &name, const t_uint &imsizex, const t_uint &imsizey
   auto const Psi = sopt::linear_transform<t_complex>(sara, imsizey, imsizex);
   const Vector<> dimage = (measurements_transform->adjoint() * uv_data.vis).real();
   Matrix<t_complex> point = Matrix<t_complex>::Zero(imsizey, imsizex);
-  point(std::floor(imsizey / 2), std::floor(imsizex / 2)) = 1.;
+  point(int(imsizey / 2), int(imsizex / 2)) = 1.;
   const Vector<> psf =
       (measurements_transform->adjoint() *
        (*measurements_transform * Vector<t_complex>::Map(point.data(), point.size())).eval())

diff --git a/cpp/example/plot_wkernel.cc b/cpp/example/plot_wkernel.cc
@@ -2,7 +2,6 @@
 #include "purify/config.h"
 
 #include <iostream>
-#include "catch.hpp"
 #include "purify/directories.h"
 
 #include "purify/types.h"