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

fmt: formatting improvements

Browse Source
This commit is contained in:
Deepak Mallubhotla 2023-07-24 01:55:37 -05:00
parent 33cab9ab41
commit fccf50eb27
Signed by: deepak
GPG Key ID: BEBAEBF28083E022
3 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
deepdog/__init__.py
View File

@ -6,6 +6,7 @@ from deepdog.real_spectrum_run import RealSpectrumRun
from deepdog.temp_aware_real_spectrum_run import TempAwareRealSpectrumRun
from deepdog.bayes_run_with_ss import BayesRunWithSubspaceSimulation
def get_version():
return __version__

10
deepdog/bayes_run_with_ss.py
View File

@ -8,7 +8,6 @@ import pdme.util.fast_nonlocal_spectrum
from typing import Sequence, Tuple, List, Optional
import datetime
import csv
import multiprocessing
import logging
import numpy
import numpy.typing
@ -55,7 +54,7 @@ class BayesRunWithSubspaceSimulation:
filename_slug: str,
max_frequency: float = 20,
end_threshold: float = None,
run_count = 100,
run_count=100,
chunksize: int = CHUNKSIZE,
ss_n_c: int = 500,
ss_n_s: int = 100,
@ -119,7 +118,7 @@ class BayesRunWithSubspaceSimulation:
raise ValueError(
f"end_threshold should be between 0 and 1, but is actually {end_threshold}"
)
self.ss_n_c = ss_n_c
self.ss_n_s = ss_n_s
self.ss_m_max = ss_m_max
@ -145,10 +144,7 @@ class BayesRunWithSubspaceSimulation:
# Generate the actual dipoles
actual_dipoles = self.actual_model.get_dipoles(self.max_frequency)
measurements = actual_dipoles.get_dot_measurements(
self.dot_inputs
)
measurements = actual_dipoles.get_dot_measurements(self.dot_inputs)
_logger.info(f"Going to work on dipole at {actual_dipoles.dipoles}")

5
deepdog/subset_simulation/subset_simulation_impl.py
View File

@ -3,7 +3,6 @@ import numpy
import pdme.measurement
import pdme.measurement.input_types
import pdme.subspace_simulation
import bisect
from typing import Sequence, Tuple, Optional
from dataclasses import dataclass
@ -71,11 +70,11 @@ class SubsetSimulation:
self.default_w_log_step = default_w_log_step
self.default_upper_w_log_step = default_upper_w_log_step
_logger.info(f"using params:")
_logger.info("using params:")
_logger.info(f"\tn_c: {self.n_c}")
_logger.info(f"\tn_s: {self.n_s}")
_logger.info(f"\tm: {self.m_max}")
_logger.info(f"let's do level 0...")
_logger.info("let's do level 0...")
self.target_cost = target_cost
_logger.info(f"will stop at target cost {target_cost}")