boltz-vsynthes 1.0.3__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 +1 -28
- {boltz_vsynthes-1.0.3.dist-info → boltz_vsynthes-1.0.4.dist-info}/METADATA +1 -1
- {boltz_vsynthes-1.0.3.dist-info → boltz_vsynthes-1.0.4.dist-info}/RECORD +7 -7
- {boltz_vsynthes-1.0.3.dist-info → boltz_vsynthes-1.0.4.dist-info}/WHEEL +0 -0
- {boltz_vsynthes-1.0.3.dist-info → boltz_vsynthes-1.0.4.dist-info}/entry_points.txt +0 -0
- {boltz_vsynthes-1.0.3.dist-info → boltz_vsynthes-1.0.4.dist-info}/licenses/LICENSE +0 -0
- {boltz_vsynthes-1.0.3.dist-info → boltz_vsynthes-1.0.4.dist-info}/top_level.txt +0 -0
boltz/main.py
CHANGED
@@ -678,19 +678,6 @@ def process_inputs(
|
|
678
678
|
# Create a subdirectory for this input file
|
679
679
|
file_out_dir = out_dir / input_file.stem
|
680
680
|
file_out_dir.mkdir(parents=True, exist_ok=True)
|
681
|
-
|
682
|
-
# Check if records exist at output path
|
683
|
-
records_dir = file_out_dir / "processed" / "records"
|
684
|
-
if records_dir.exists():
|
685
|
-
# Load existing records
|
686
|
-
existing = [Record.load(p) for p in records_dir.glob("*.json")]
|
687
|
-
processed_ids = {record.id for record in existing}
|
688
|
-
|
689
|
-
# Skip if already processed
|
690
|
-
if input_file.stem in processed_ids:
|
691
|
-
click.echo(f"Found existing processed input for {input_file.stem}, skipping.")
|
692
|
-
all_records.extend(existing)
|
693
|
-
continue
|
694
681
|
|
695
682
|
# Create output directories for this file
|
696
683
|
msa_dir = file_out_dir / "msa"
|
@@ -952,11 +939,6 @@ def cli() -> None:
|
|
952
939
|
is_flag=True,
|
953
940
|
help="Whether to disable the kernels. Default False",
|
954
941
|
)
|
955
|
-
@click.option(
|
956
|
-
"--skip_structure",
|
957
|
-
is_flag=True,
|
958
|
-
help="Skip structure prediction and go straight to affinity prediction. Default is False.",
|
959
|
-
)
|
960
942
|
def predict( # noqa: C901, PLR0915, PLR0912
|
961
943
|
data: str,
|
962
944
|
out_dir: str,
|
@@ -990,7 +972,6 @@ def predict( # noqa: C901, PLR0915, PLR0912
|
|
990
972
|
subsample_msa: bool = True,
|
991
973
|
num_subsampled_msa: int = 1024,
|
992
974
|
no_kernels: bool = False,
|
993
|
-
skip_structure: bool = False,
|
994
975
|
) -> None:
|
995
976
|
"""Run predictions with Boltz."""
|
996
977
|
# If cpu, write a friendly warning
|
@@ -1151,7 +1132,7 @@ def predict( # noqa: C901, PLR0915, PLR0912
|
|
1151
1132
|
precision=32 if model == "boltz1" else "bf16-mixed",
|
1152
1133
|
)
|
1153
1134
|
|
1154
|
-
if filtered_manifest.records
|
1135
|
+
if filtered_manifest.records:
|
1155
1136
|
msg = f"Running structure prediction for {len(filtered_manifest.records)} input"
|
1156
1137
|
msg += "s." if len(filtered_manifest.records) > 1 else "."
|
1157
1138
|
click.echo(msg)
|
@@ -1240,14 +1221,6 @@ def predict( # noqa: C901, PLR0915, PLR0912
|
|
1240
1221
|
msg += "s." if len(manifest_filtered.records) > 1 else "."
|
1241
1222
|
click.echo(msg)
|
1242
1223
|
|
1243
|
-
# When skipping structure prediction, we need to ensure structure files exist
|
1244
|
-
if skip_structure:
|
1245
|
-
for record in manifest_filtered.records:
|
1246
|
-
structure_path = out_dir / "predictions" / record.id / f"pre_affinity_{record.id}.npz"
|
1247
|
-
if not structure_path.exists():
|
1248
|
-
msg = f"Structure file not found for {record.id}. Cannot run affinity prediction without structure."
|
1249
|
-
raise FileNotFoundError(msg)
|
1250
|
-
|
1251
1224
|
pred_writer = BoltzAffinityWriter(
|
1252
1225
|
data_dir=processed.targets_dir,
|
1253
1226
|
output_dir=out_dir / "predictions",
|
@@ -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
|