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

feat: add configurable max number of dipoles to write
All checks were successful
gitea-physics/deepdog/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Deepak Mallubhotla 2024-12-29 21:14:59 -06:00
parent 53f8993f2b
commit a1b59cd18b
Signed by: deepak
GPG Key ID: BEBAEBF28083E022
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
deepdog/direct_monte_carlo/direct_mc.py
View File

@ -287,7 +287,7 @@ class DirectMonteCarloRun:
_logger.debug(f"Pool results: {pool_results}") _logger.debug(f"Pool results: {pool_results}")
if self.config.write_successes_to_file: if self.config.write_successes_to_file:
_logger.info("Writing dipole results") _logger.info("Writing dipole results")
cycle_success_configs = numpy.concatenate( cycle_success_configs = numpy.concatenate(
@ -301,23 +301,27 @@ class DirectMonteCarloRun:
f"Limiting to {max_number_dipoles_to_write=}, have {len(cycle_success_configs)}" f"Limiting to {max_number_dipoles_to_write=}, have {len(cycle_success_configs)}"
) )
if len(cycle_success_configs): if len(cycle_success_configs):
sorted_by_freq = numpy.array( sorted_by_freq = numpy.array(
[ [
pdme.subspace_simulation.sort_array_of_dipoles_by_frequency( pdme.subspace_simulation.sort_array_of_dipoles_by_frequency(
dipole_config dipole_config
) )
for dipole_config in cycle_success_configs[:max_number_dipoles_to_write] for dipole_config in cycle_success_configs[
:max_number_dipoles_to_write
]
] ]
) )
for n in range(dipole_count): for n in range(dipole_count):
dipole_filename = f"{self.config.tag}_{step_count}_dipole_{n}.csv" dipole_filename = (
_logger.debug(f"Writing {min(len(cycle_success_configs), max_number_dipoles_to_write)} to {dipole_filename}") f"{self.config.tag}_{step_count}_dipole_{n}.csv"
)
_logger.debug(
f"Writing {min(len(cycle_success_configs), max_number_dipoles_to_write)} to {dipole_filename}"
)
numpy.savetxt( numpy.savetxt(
dipole_filename, dipole_filename,
sorted_by_freq[:, n], sorted_by_freq[:, n],