fix: update compare_counts.py to user newer pyroe

COMBINE-lab · Jun 1, 2022 · 9a812a7 · 9a812a7
1 parent 7b9204c
commit 9a812a7
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/scripts/testing/compare_counts.py b/scripts/testing/compare_counts.py
@@ -50,8 +50,8 @@ def compare_quants(args):
 	usa_mode = mode_ng_ref[0]
 	odict = None
 	if usa_mode:
-		a = load_fry.load_fry(ref_quant_dir, output_format='raw')
-		b = load_fry.load_fry(test_quant_dir, output_format='raw')
+		a = load_fry(ref_quant_dir, output_format='raw')
+		b = load_fry(test_quant_dir, output_format='raw')
 
 		odict = { "nobs_ref" : a.n_obs, "nobs_test" : b.n_obs }
 
@@ -62,8 +62,8 @@ def compare_quants(args):
 		odict['obs_ref-obs_test'] = list(a.obs_names.difference(b.obs_names))
 		odict['obs_test-obs_ref'] = list(b.obs_names.difference(a.obs_names))
 	else:
-		a = load_fry.load_fry(ref_quant_dir, output_format='raw')
-		b = load_fry.load_fry(test_quant_dir, output_format='raw')
+		a = load_fry(ref_quant_dir, output_format='raw')
+		b = load_fry(test_quant_dir, output_format='raw')
 
 		odict = { "nobs_ref" : a.n_obs, "nobs_test" : b.n_obs }
 
@@ -87,4 +87,4 @@ def compare_quants(args):
 	parser.add_argument('output', type=str,
 		help='Where to write the output report')
 	args = parser.parse_args()
-	compare_quants(args)
+	compare_quants(args)