Script to standalone application of flows

CaioDaumann · Jan 28, 2025 · 9bd94aa · 9bd94aa
1 parent dd82119
commit 9bd94aa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/post_processing_script/apply_flow_to_parquet.py b/post_processing_script/apply_flow_to_parquet.py
@@ -47,7 +47,7 @@ def main():
 
         # Reading the files as a dataframe
         files = glob.glob( args.mcfilespath ) #+ systematic + "/*.parquet" )
-        files = files[:150]
+        #files = files[:150]
         if len(files) == 0:
             print( "\nSystematic ", systematic , " is not avaliable in the path: ", args.mcfilespath  )
             continue
@@ -158,7 +158,7 @@ def main():
                 mc_df[ "sigma_m_over_m_smeared_corr" ] = utils.calculate_corrected_smeared_sigma_m_over_m(mc_df)
 
         # Dumping the df file with the new entries
-        mc_df.to_parquet( args.outpath + "/Zmmg_out_Fix_AR.parquet")
+        mc_df.to_parquet( args.outpath + "/Zee_with_flows.parquet")
         print('\n\nFinished!\n')
         exit()
 

diff --git a/post_processing_script/run_flow_applier.sh b/post_processing_script/run_flow_applier.sh
@@ -3,7 +3,7 @@
 #python apply_flow_to_parquet.py --mcfilespath "./validate_post_processing_script/jq_zmmg_out.parquet" --process "Zmmg" --period "postEE" --outpath "./validate_post_processing_script/"
 
 # For paper with IC guys !!!! (Zee)
-#python apply_flow_to_parquet.py --mcfilespath "/net/scratch_cms3a/daumann/normalizing_flows_project/script_to_prepare_samples_for_paper/splited_parquet/DY_test.parquet" --process "Zee" --period "postEE" --outpath "./paper_with_IC/"
+python apply_flow_to_parquet.py --mcfilespath "/net/scratch_cms3a/daumann/normalizing_flows_project/script_to_prepare_samples_for_paper/splited_parquet/DY_test.parquet" --process "Zee" --period "postEE" --outpath "./paper_with_IC/"
 
 # for zmmg
-python apply_flow_to_parquet.py --mcfilespath "/net/scratch_cms3a/daumann/massresdecorrhiggsdna/sigma_syst/jon_high_stat_zmmg/" --process "Zmmg" --period "postEE" --outpath "./paper_with_IC/"
+#python apply_flow_to_parquet.py --mcfilespath "/net/scratch_cms3a/daumann/massresdecorrhiggsdna/sigma_syst/jon_high_stat_zmmg/" --process "Zmmg" --period "postEE" --outpath "./paper_with_IC/"