From fb018abeae2adf4438a030140a6c905f11bb6bc1 Mon Sep 17 00:00:00 2001 From: Deepak Mallubhotla Date: Wed, 1 May 2024 21:40:53 -0500 Subject: [PATCH] feat: allows disabling timestamps in directmc bayesrun files --- deepdog/direct_monte_carlo/direct_mc.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/deepdog/direct_monte_carlo/direct_mc.py b/deepdog/direct_monte_carlo/direct_mc.py index f77a0f4..3bbabde 100644 --- a/deepdog/direct_monte_carlo/direct_mc.py +++ b/deepdog/direct_monte_carlo/direct_mc.py @@ -37,6 +37,7 @@ class DirectMonteCarloConfig: cap_core_count: int = 0 # 0 means cap at num cores - 1 chunk_size: int = 50 write_bayesrun_file = True + bayesrun_file_timestamp = True # chunk size of some kind @@ -284,9 +285,14 @@ class DirectMonteCarloRun: if self.config.write_bayesrun_file: + if self.config.bayesrun_file_timestamp: + timestamp_str = f"{timestamp}-" + else: + timestamp_str = "" 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}]") # row: Dict[str, Union[int, float, str]] = {} row = {}