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 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 and not skip_structure:
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,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: boltz-vsynthes
3
- Version: 1.0.2
3
+ Version: 1.0.4
4
4
  Summary: Boltz for V-Synthes
5
5
  Requires-Python: <3.13,>=3.10
6
6
  Description-Content-Type: text/markdown
@@ -1,5 +1,5 @@
1
1
  boltz/__init__.py,sha256=F_-so3S40iZrSZ89Ge4TS6aZqwWyZXq_H4AXGDlbA_g,187
2
- boltz/main.py,sha256=jesu0Y9aBVNPjulshsfyg78HTGNh2K9YjF76rWUWuFs,40833
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.2.dist-info/licenses/LICENSE,sha256=8GZ_1eZsUeG6jdqgJJxtciWzADfgLEV4LY8sKUOsJhc,1102
108
- boltz_vsynthes-1.0.2.dist-info/METADATA,sha256=jjCm6Og0LpIvWLHgHJLnrHNkaGMO4cqZYf73yLKk8rM,7171
109
- boltz_vsynthes-1.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
110
- boltz_vsynthes-1.0.2.dist-info/entry_points.txt,sha256=n5a5I35ntu9lmyr16oZgHPFY0b0YxjiixY7m7nbMTLc,41
111
- boltz_vsynthes-1.0.2.dist-info/top_level.txt,sha256=MgU3Jfb-ctWm07YGMts68PMjSh9v26D0gfG3dFRmVFA,6
112
- boltz_vsynthes-1.0.2.dist-info/RECORD,,
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,,