Update regression tests and defaults

This commit is contained in:
jbutch
2025-11-23 20:21:01 -08:00
parent 12115d052b
commit 22407d510c
8 changed files with 6 additions and 6 deletions

View File

@@ -179,7 +179,7 @@ class DesignInputSpecification(BaseModel):
infer_ori_strategy: Optional[str] = Field(None, description="Strategy for inferring origin; `com` or `hotspots`")
# Additional global conditioning
plddt_enhanced: Optional[bool] = Field(True, description="Enable pLDDT enhancement")
is_non_loopy: Optional[bool] = Field(True, description="Non-loopy conditioning")
is_non_loopy: Optional[bool] = Field(None, description="Non-loopy conditioning")
# Partial diffusion
partial_t: Optional[float] = Field(None, ge=0.0, description="Angstroms of noise to add for partial diffusion (None turns off partial diffusion), t <= 15 recommended.")
# fmt: on

View File

@@ -429,7 +429,7 @@ def create_atom_array_from_design_specification_legacy(
symmetry: dict = None,
# Low-temperature global conditioning args
plddt_enhanced: bool = True,
is_non_loopy: bool | None = True,
is_non_loopy: bool | None = None,
# Partial diff args:
partial_t: float | None = None, # Optional noise scale for partial diffusion
**_, # dump additional args

View File

@@ -180,10 +180,10 @@ def test_atom14_pipeline_regression(
# Get regression data path using shared logic
regression_path = _get_regression_data_path(example_name, is_inference, config.name)
# Uncomment the following lines to create/update the regression data
with regression_path.open("wb") as f:
pickle.dump(result, f)
logger.info(f"Saved regression data to {regression_path}")
# # Uncomment the following lines to create/update the regression data
# with regression_path.open("wb") as f:
# pickle.dump(result, f)
# logger.info(f"Saved regression data to {regression_path}")
# Load expected result
with regression_path.open("rb") as f: