physics/kalpa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: fixes for override measurement file finding

Browse Source
This commit is contained in:
Deepak Mallubhotla 2025-02-23 23:30:19 -06:00
parent 31ba558364
commit 5ad98f01fe
Signed by: deepak
GPG Key ID: BEBAEBF28083E022
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
kalpaa/stages/stage01.py
View File

@ -201,9 +201,7 @@ class Stage01Runner:
def run(self): def run(self):
seed_index = 0 seed_index = 0
if self.config.generation_config.override_measurement_filesets is not None: if self.config.generation_config.override_measurement_filesets is not None:
for ( for override_name in self.config.generation_config.override_measurement_filesets.keys():
override_name,
) in self.config.generation_config.override_measurement_filesets.keys():
# don't need to do anything with the files, just create the out dir # don't need to do anything with the files, just create the out dir
out = self.config.get_out_dir_path() out = self.config.get_out_dir_path()
directory = out / f"{override_name}" directory = out / f"{override_name}"

4
kalpaa/stages/stage02.py
View File

@ -115,7 +115,7 @@ class Stage02Runner:
_logger.info(f"Working on {trial_name=}") _logger.info(f"Working on {trial_name=}")
_logger.debug(f"Have {seed_index=}") _logger.debug(f"Have {seed_index=}")
self.single_run_in_subdir( self.single_run_in_subdir(
job_index, cost, dot.label, trial_name, seed_index job_index, cost, dot.label, trial_name, seed_index, override_name=override_key
) )
def single_run_in_subdir( def single_run_in_subdir(
@ -147,7 +147,7 @@ class Stage02Runner:
"override_name provided but no override_measurement_filesets, shouldn't be possible to get here" "override_name provided but no override_measurement_filesets, shouldn't be possible to get here"
) )
_logger.info(f"Time to read override measurement fileset {override_name}") _logger.info(f"Time to read override measurement fileset {override_name}")
override_dir = self.config.get_override_dir_path() override_dir = self.config.get_override_dir_path() / override_name
override_measurements = ( override_measurements = (
self.config.generation_config.override_measurement_filesets[ self.config.generation_config.override_measurement_filesets[
override_name override_name