boltz-vsynthes 1.0.2__py3-none-any.whl → 1.0.4__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- boltz/main.py +6 -23
- {boltz_vsynthes-1.0.2.dist-info → boltz_vsynthes-1.0.4.dist-info}/METADATA +1 -1
- {boltz_vsynthes-1.0.2.dist-info → boltz_vsynthes-1.0.4.dist-info}/RECORD +7 -7
- {boltz_vsynthes-1.0.2.dist-info → boltz_vsynthes-1.0.4.dist-info}/WHEEL +0 -0
- {boltz_vsynthes-1.0.2.dist-info → boltz_vsynthes-1.0.4.dist-info}/entry_points.txt +0 -0
- {boltz_vsynthes-1.0.2.dist-info → boltz_vsynthes-1.0.4.dist-info}/licenses/LICENSE +0 -0
- {boltz_vsynthes-1.0.2.dist-info → boltz_vsynthes-1.0.4.dist-info}/top_level.txt +0 -0
boltz/main.py
CHANGED
@@ -669,29 +669,19 @@ def process_inputs(
|
|
669
669
|
processed_dir = out_dir / "processed"
|
670
670
|
processed_dir.mkdir(parents=True, exist_ok=True)
|
671
671
|
|
672
|
+
# Create structure directory
|
673
|
+
structure_dir = processed_dir / "structures"
|
674
|
+
structure_dir.mkdir(parents=True, exist_ok=True)
|
675
|
+
|
672
676
|
# Process each input file in its own directory
|
673
677
|
for input_file in data:
|
674
678
|
# Create a subdirectory for this input file
|
675
679
|
file_out_dir = out_dir / input_file.stem
|
676
680
|
file_out_dir.mkdir(parents=True, exist_ok=True)
|
677
|
-
|
678
|
-
# Check if records exist at output path
|
679
|
-
records_dir = file_out_dir / "processed" / "records"
|
680
|
-
if records_dir.exists():
|
681
|
-
# Load existing records
|
682
|
-
existing = [Record.load(p) for p in records_dir.glob("*.json")]
|
683
|
-
processed_ids = {record.id for record in existing}
|
684
|
-
|
685
|
-
# Skip if already processed
|
686
|
-
if input_file.stem in processed_ids:
|
687
|
-
click.echo(f"Found existing processed input for {input_file.stem}, skipping.")
|
688
|
-
all_records.extend(existing)
|
689
|
-
continue
|
690
681
|
|
691
682
|
# Create output directories for this file
|
692
683
|
msa_dir = file_out_dir / "msa"
|
693
684
|
records_dir = file_out_dir / "processed" / "records"
|
694
|
-
structure_dir = file_out_dir / "processed" / "structures"
|
695
685
|
processed_msa_dir = file_out_dir / "processed" / "msa"
|
696
686
|
processed_constraints_dir = file_out_dir / "processed" / "constraints"
|
697
687
|
processed_templates_dir = file_out_dir / "processed" / "templates"
|
@@ -701,7 +691,6 @@ def process_inputs(
|
|
701
691
|
file_out_dir.mkdir(parents=True, exist_ok=True)
|
702
692
|
msa_dir.mkdir(parents=True, exist_ok=True)
|
703
693
|
records_dir.mkdir(parents=True, exist_ok=True)
|
704
|
-
structure_dir.mkdir(parents=True, exist_ok=True)
|
705
694
|
processed_msa_dir.mkdir(parents=True, exist_ok=True)
|
706
695
|
processed_constraints_dir.mkdir(parents=True, exist_ok=True)
|
707
696
|
processed_templates_dir.mkdir(parents=True, exist_ok=True)
|
@@ -730,7 +719,7 @@ def process_inputs(
|
|
730
719
|
processed_constraints_dir=processed_constraints_dir,
|
731
720
|
processed_templates_dir=processed_templates_dir,
|
732
721
|
processed_mols_dir=processed_mols_dir,
|
733
|
-
structure_dir=structure_dir,
|
722
|
+
structure_dir=structure_dir, # Use the central structure directory
|
734
723
|
records_dir=records_dir,
|
735
724
|
)
|
736
725
|
|
@@ -950,11 +939,6 @@ def cli() -> None:
|
|
950
939
|
is_flag=True,
|
951
940
|
help="Whether to disable the kernels. Default False",
|
952
941
|
)
|
953
|
-
@click.option(
|
954
|
-
"--skip_structure",
|
955
|
-
is_flag=True,
|
956
|
-
help="Skip structure prediction and go straight to affinity prediction. Default is False.",
|
957
|
-
)
|
958
942
|
def predict( # noqa: C901, PLR0915, PLR0912
|
959
943
|
data: str,
|
960
944
|
out_dir: str,
|
@@ -988,7 +972,6 @@ def predict( # noqa: C901, PLR0915, PLR0912
|
|
988
972
|
subsample_msa: bool = True,
|
989
973
|
num_subsampled_msa: int = 1024,
|
990
974
|
no_kernels: bool = False,
|
991
|
-
skip_structure: bool = False,
|
992
975
|
) -> None:
|
993
976
|
"""Run predictions with Boltz."""
|
994
977
|
# If cpu, write a friendly warning
|
@@ -1149,7 +1132,7 @@ def predict( # noqa: C901, PLR0915, PLR0912
|
|
1149
1132
|
precision=32 if model == "boltz1" else "bf16-mixed",
|
1150
1133
|
)
|
1151
1134
|
|
1152
|
-
if filtered_manifest.records
|
1135
|
+
if filtered_manifest.records:
|
1153
1136
|
msg = f"Running structure prediction for {len(filtered_manifest.records)} input"
|
1154
1137
|
msg += "s." if len(filtered_manifest.records) > 1 else "."
|
1155
1138
|
click.echo(msg)
|
@@ -1,5 +1,5 @@
|
|
1
1
|
boltz/__init__.py,sha256=F_-so3S40iZrSZ89Ge4TS6aZqwWyZXq_H4AXGDlbA_g,187
|
2
|
-
boltz/main.py,sha256=
|
2
|
+
boltz/main.py,sha256=ZNaNplHLMr9OmLMhbivgducwbGCrn8g67WRPpuhh9GA,40084
|
3
3
|
boltz/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
4
4
|
boltz/data/const.py,sha256=1M-88Z6HkfKY6MkNtqcj3b9P-oX9xEXluh3qM_u8dNU,26779
|
5
5
|
boltz/data/mol.py,sha256=maOpPHEGX1VVXCIFY6pQNGF7gUBZPAfgSvuPf2QO1yc,34268
|
@@ -104,9 +104,9 @@ boltz/model/optim/scheduler.py,sha256=nB4jz0CZ4pR4n08LQngExL_pNycIdYI8AXVoHPnZWQ
|
|
104
104
|
boltz/model/potentials/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
105
105
|
boltz/model/potentials/potentials.py,sha256=vev8Vjfs-ML1hyrdv_R8DynG4wSFahJ6nzPWp7CYQqw,17507
|
106
106
|
boltz/model/potentials/schedules.py,sha256=m7XJjfuF9uTX3bR9VisXv1rvzJjxiD8PobXRpcBBu1c,968
|
107
|
-
boltz_vsynthes-1.0.
|
108
|
-
boltz_vsynthes-1.0.
|
109
|
-
boltz_vsynthes-1.0.
|
110
|
-
boltz_vsynthes-1.0.
|
111
|
-
boltz_vsynthes-1.0.
|
112
|
-
boltz_vsynthes-1.0.
|
107
|
+
boltz_vsynthes-1.0.4.dist-info/licenses/LICENSE,sha256=8GZ_1eZsUeG6jdqgJJxtciWzADfgLEV4LY8sKUOsJhc,1102
|
108
|
+
boltz_vsynthes-1.0.4.dist-info/METADATA,sha256=5oK6_h_oPquPVFGAwT9rRFOd7F9RJ8jrC7UGKR8tRVQ,7171
|
109
|
+
boltz_vsynthes-1.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
110
|
+
boltz_vsynthes-1.0.4.dist-info/entry_points.txt,sha256=n5a5I35ntu9lmyr16oZgHPFY0b0YxjiixY7m7nbMTLc,41
|
111
|
+
boltz_vsynthes-1.0.4.dist-info/top_level.txt,sha256=MgU3Jfb-ctWm07YGMts68PMjSh9v26D0gfG3dFRmVFA,6
|
112
|
+
boltz_vsynthes-1.0.4.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|