physics/deepdog
2
0
Fork 0
Code Issues 2 Pull Requests 9 Projects Releases Wiki Activity

feat: allows disabling timestamps in directmc bayesrun files
All checks were successful
gitea-physics/deepdog/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Deepak Mallubhotla 2024-05-01 21:40:53 -05:00
parent d28c190816
commit fb018abeae
Signed by: deepak
GPG Key ID: 976F3357369149AB
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
deepdog/direct_monte_carlo/direct_mc.py
View File

@ -37,6 +37,7 @@ class DirectMonteCarloConfig:
cap_core_count: int = 0 # 0 means cap at num cores - 1 cap_core_count: int = 0 # 0 means cap at num cores - 1
chunk_size: int = 50 chunk_size: int = 50
write_bayesrun_file = True write_bayesrun_file = True
bayesrun_file_timestamp = True
# chunk size of some kind # chunk size of some kind
@ -284,9 +285,14 @@ class DirectMonteCarloRun:
if self.config.write_bayesrun_file: if self.config.write_bayesrun_file:
if self.config.bayesrun_file_timestamp:
timestamp_str = f"{timestamp}-"
else:
timestamp_str = ""
filename = ( filename = (
f"{timestamp}-{self.config.tag}.realdata.fast_filter.bayesrun.csv" f"{timestamp_str}{self.config.tag}.realdata.fast_filter.bayesrun.csv"
) )
_logger.info(f"Going to write to file [{filename}]") _logger.info(f"Going to write to file [{filename}]")
# row: Dict[str, Union[int, float, str]] = {} # row: Dict[str, Union[int, float, str]] = {}
row = {} row = {}