debug

Author: AstroPatty

.github/workflows/test.yaml

@@ -116,4 +116,4 @@ jobs:
         run: echo "Running on $(nproc) cores"
       - name: Run tests with pytest
         run: |
-          uv run mpiexec --timeout 120 --use-hwthread-cpus -n 4 pytest -m parallel test/parallel -x
+          uv run mpiexec --timeout 120 --use-hwthread-cpus -n 4 pytest -m parallel test/parallel/test_lc_mpi.py::test_write_some_missing -s

src/opencosmo/io/writers.py

@@ -85,6 +85,7 @@ def write(self, file: h5py.File | h5py.Group):
         child_names = list(self.children.keys())
         child_names.sort()
         for name in child_names:
+            print(f"Writing dataset {name}")
             self.children[name].write(file[name])
 
 
@@ -114,7 +115,9 @@ def write(self, group: h5py.Group):
         names = list(self.columns.keys())
         names.sort()
         for colname in names:
+            print(f"Writing column {colname}")
             self.columns[colname].write(data_group)
+        print("writing links")
         if self.links:
             link_group = group["data_linked"]
             link_names = list(self.links.keys())
@@ -123,9 +126,11 @@ def write(self, group: h5py.Group):
             for name in link_names:
                 self.links[name].write(link_group)
 
+        print("writing spatial index")
         if self.spatial_index is not None:
             index_group = group["index"]
             self.spatial_index.write(index_group)
+        print("done")
 
 
 class EmptyColumnWriter: