Add assert for chunk files

PMBio · Apr 30, 2024 · 4418062 · 4418062
1 parent 88d6dc8
commit 4418062
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deeprvat/deeprvat/associate.py b/deeprvat/deeprvat/associate.py
@@ -959,7 +959,9 @@ def compute_burdens(
     burden_files = ["burdens.zarr", "x.zarr", "y.zarr", "sample_ids.zarr"]
 
     for burden_file in burden_files:
-        assert (Path(out_dir) / f"chunk_{chunk}" / burden_file).exists()
+        burden_file_path = Path(out_dir) / "chunks" / f"chunk_{chunk}" / burden_file
+        logger.debug(f"Checking if {burden_file_path} exists")
+        assert burden_file_path.exists()
     else:
         logger.info(f"All zarr files exists for chunk {chunk}")