mirror of
https://github.com/RosettaCommons/foundry.git
synced 2026-06-04 13:24:22 +08:00
Update regression tests and defaults
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user