tests pass in intermediate fake dataset state.

MPoL-dev · Jan 1, 2024 · 7e55d0f · 7e55d0f
1 parent 53ec3ad
commit 7e55d0f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/test/conftest.py b/test/conftest.py
@@ -37,7 +37,7 @@ def packed_cube(img2D_butterfly):
     """Create a packed tensor image cube from the butterfly."""
     # now (1, npix, npix)
     print("npix packed cube", img2D_butterfly.shape)
-    nchan = 10
+    nchan = 9
     # tile to some nchan, npix, npix
     cube = torch.tile(torch.from_numpy(img2D_butterfly), (nchan, 1, 1))
     # convert to packed format

diff --git a/test/losses_test.py b/test/losses_test.py
@@ -107,6 +107,7 @@ def test_rchi_evaluation(
     print("loose nll", log_like)
 
     # calculate the gridded log likelihood
+    print(gridded_vis_model.size(), dataset.mask.size())
     log_like_gridded = losses.r_chi_squared_gridded(gridded_vis_model, dataset)
     print("gridded nll", log_like_gridded)