fmt: formatting improvements

This commit is contained in:
2023-07-24 01:55:37 -05:00
parent 33cab9ab41
commit fccf50eb27
3 changed files with 6 additions and 10 deletions

View File

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

View File

@@ -8,7 +8,6 @@ import pdme.util.fast_nonlocal_spectrum
from typing import Sequence, Tuple, List, Optional from typing import Sequence, Tuple, List, Optional
import datetime import datetime
import csv import csv
import multiprocessing
import logging import logging
import numpy import numpy
import numpy.typing import numpy.typing
@@ -145,10 +144,7 @@ class BayesRunWithSubspaceSimulation:
# Generate the actual dipoles # Generate the actual dipoles
actual_dipoles = self.actual_model.get_dipoles(self.max_frequency) actual_dipoles = self.actual_model.get_dipoles(self.max_frequency)
measurements = actual_dipoles.get_dot_measurements( measurements = actual_dipoles.get_dot_measurements(self.dot_inputs)
self.dot_inputs
)
_logger.info(f"Going to work on dipole at {actual_dipoles.dipoles}") _logger.info(f"Going to work on dipole at {actual_dipoles.dipoles}")

View File

@@ -3,7 +3,6 @@ import numpy
import pdme.measurement import pdme.measurement
import pdme.measurement.input_types import pdme.measurement.input_types
import pdme.subspace_simulation import pdme.subspace_simulation
import bisect
from typing import Sequence, Tuple, Optional from typing import Sequence, Tuple, Optional
from dataclasses import dataclass from dataclasses import dataclass
@@ -71,11 +70,11 @@ class SubsetSimulation:
self.default_w_log_step = default_w_log_step self.default_w_log_step = default_w_log_step
self.default_upper_w_log_step = default_upper_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_c: {self.n_c}")
_logger.info(f"\tn_s: {self.n_s}") _logger.info(f"\tn_s: {self.n_s}")
_logger.info(f"\tm: {self.m_max}") _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 self.target_cost = target_cost
_logger.info(f"will stop at target cost {target_cost}") _logger.info(f"will stop at target cost {target_cost}")