list unpacking and file reading updated

HALFpipe · Apr 19, 2024 · a363eca · a363eca
1 parent dc35152
commit a363eca
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/src/gwas/src/gwas/plots/cli.py b/src/gwas/src/gwas/plots/cli.py
@@ -74,7 +74,7 @@ def main():
 
     input_dir = args.input_directory
 
-    verify_metadata(input_dir)
+    verify_metadata(*input_dir)
     resolved_chrs = resolve_chromosomes(input_dir, logger=logger)
 
     phenotype_stat_names = enrich_phenotype_names(args.include_phenotype_list)

diff --git a/src/gwas/src/gwas/plots/helpers.py b/src/gwas/src/gwas/plots/helpers.py
@@ -25,12 +25,13 @@ def chi2_pvalue(ustat: int | float, vstat: int | float) -> float:
 def load_metadata(metadata_path: Path):
     with CompressedBytesReader(metadata_path) as f:
         decompressor = zstd.ZstdDecompressor()
-        compressed_data = f.read()
+        # compressed_data = f.read()
         # decompressed_data = decompressor.decompress(compressed_data)
-        stream_reader = decompressor.stream_reader(compressed_data)
-        decompressed_data = stream_reader.read()
-        stream_reader.close()
-        metadata = pickle.loads(decompressed_data)
+        # stream_reader = decompressor.stream_reader(compressed_data)
+        with decompressor.stream_reader(f) as reader:
+            decompressed_data = reader.read()
+            reader.close()
+            metadata = pickle.loads(decompressed_data)
     return metadata