Skip to content

Commit

Permalink
update configs to remove low memory metrics
Browse files Browse the repository at this point in the history
  • Loading branch information
geoffwoollard committed Sep 18, 2024
1 parent 1f4057e commit b33a6a0
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 18 deletions.
1 change: 1 addition & 0 deletions src/cryo_challenge/data/_validation/config_validators.py
Original file line number Diff line number Diff line change
Expand Up @@ -151,6 +151,7 @@ def validate_config_mtm_analysis(config_analysis: dict) -> None:
chunk_size_submission: int, is the chunk size for the submission volume.
chunk_size_gt: int, is the chunk size for the ground truth volume.
normalize: dict, is the normalize part of the analysis part of the config.
low_memory: dict, is the low memory part of the analysis part of the config. # TODO: add validation for low_memory
""" # noqa: E501
keys_and_types = {
Expand Down
18 changes: 0 additions & 18 deletions src/cryo_challenge/data/_validation/output_validators.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,27 +19,18 @@ class MapToMapResultsValidator:
config: dict, input config dictionary.
user_submitted_populations: torch.Tensor, user submitted populations, which sum to 1.
corr: dict, correlation results.
corr_low_memory: dict, correlation results in low memory mode.
l2: dict, L2 results.
l2_low_memory: dict, L2 results in low memory mode.
bioem: dict, BioEM results.
bioem_low_memory: dict, BioEM results in low memory mode.
fsc: dict, FSC results.
fsc_low_memory: dict, FSC results in low memory mode.
"""

config: dict
user_submitted_populations: torch.Tensor
corr: Optional[dict] = None
corr_low_memory: Optional[dict] = None
l2: Optional[dict] = None
l2_low_memory: Optional[dict] = None
bioem: Optional[dict] = None
bioem_low_memory: Optional[dict] = None
fsc: Optional[dict] = None
fsc_low_memory: Optional[dict] = None
res: Optional[dict] = None
res_low_memory: Optional[dict] = None

def __post_init__(self):
validate_input_config_mtm(self.config)
Expand Down Expand Up @@ -150,10 +141,6 @@ class DistributionToDistributionResultsValidator:
bioem: dict, BioEM distance results.
l2: dict, L2 distance results.
corr: dict, correlation distance results.
fsc_low_memory: dict, FSC distance results in low memory mode.
bioem_low_memory: dict, BioEM distance results in low memory mode.
l2_low_memory: dict, L2 distance results in low memory mode.
corr_low_memory: dict, correlation distance results in low memory mode.
"""

config: dict
Expand All @@ -164,11 +151,6 @@ class DistributionToDistributionResultsValidator:
res: Optional[dict] = None
l2: Optional[dict] = None
corr: Optional[dict] = None
fsc_low_memory: Optional[dict] = None
bioem_low_memory: Optional[dict] = None
res_low_memory: Optional[dict] = None
l2_low_memory: Optional[dict] = None
corr_low_memory: Optional[dict] = None

def __post_init__(self):
validate_input_config_disttodist(self.config)
Expand Down

0 comments on commit b33a6a0

Please sign in to comment.