boltz-vsynthes 1.0.2__py3-none-any.whl → 1.0.4__py3-none-any.whl

boltz/main.py

@@ -669,29 +669,19 @@ def process_inputs(
     processed_dir = out_dir / "processed"
     processed_dir.mkdir(parents=True, exist_ok=True)
     
+    # Create structure directory
+    structure_dir = processed_dir / "structures"
+    structure_dir.mkdir(parents=True, exist_ok=True)
+    
     # Process each input file in its own directory
     for input_file in data:
         # Create a subdirectory for this input file
         file_out_dir = out_dir / input_file.stem
         file_out_dir.mkdir(parents=True, exist_ok=True)
-        
-        # Check if records exist at output path
-        records_dir = file_out_dir / "processed" / "records"
-        if records_dir.exists():
-            # Load existing records
-            existing = [Record.load(p) for p in records_dir.glob("*.json")]
-            processed_ids = {record.id for record in existing}
-            
-            # Skip if already processed
-            if input_file.stem in processed_ids:
-                click.echo(f"Found existing processed input for {input_file.stem}, skipping.")
-                all_records.extend(existing)
-                continue
 
         # Create output directories for this file
         msa_dir = file_out_dir / "msa"
         records_dir = file_out_dir / "processed" / "records"
-        structure_dir = file_out_dir / "processed" / "structures"
         processed_msa_dir = file_out_dir / "processed" / "msa"
         processed_constraints_dir = file_out_dir / "processed" / "constraints"
         processed_templates_dir = file_out_dir / "processed" / "templates"
@@ -701,7 +691,6 @@ def process_inputs(
         file_out_dir.mkdir(parents=True, exist_ok=True)
         msa_dir.mkdir(parents=True, exist_ok=True)
         records_dir.mkdir(parents=True, exist_ok=True)
-        structure_dir.mkdir(parents=True, exist_ok=True)
         processed_msa_dir.mkdir(parents=True, exist_ok=True)
         processed_constraints_dir.mkdir(parents=True, exist_ok=True)
         processed_templates_dir.mkdir(parents=True, exist_ok=True)
@@ -730,7 +719,7 @@ def process_inputs(
             processed_constraints_dir=processed_constraints_dir,
             processed_templates_dir=processed_templates_dir,
             processed_mols_dir=processed_mols_dir,
-            structure_dir=structure_dir,
+            structure_dir=structure_dir,  # Use the central structure directory
             records_dir=records_dir,
         )
 
@@ -950,11 +939,6 @@ def cli() -> None:
     is_flag=True,
     help="Whether to disable the kernels. Default False",
 )
-@click.option(
-    "--skip_structure",
-    is_flag=True,
-    help="Skip structure prediction and go straight to affinity prediction. Default is False.",
-)
 def predict(  # noqa: C901, PLR0915, PLR0912
     data: str,
     out_dir: str,
@@ -988,7 +972,6 @@ def predict(  # noqa: C901, PLR0915, PLR0912
     subsample_msa: bool = True,
     num_subsampled_msa: int = 1024,
     no_kernels: bool = False,
-    skip_structure: bool = False,
 ) -> None:
     """Run predictions with Boltz."""
     # If cpu, write a friendly warning
@@ -1149,7 +1132,7 @@ def predict(  # noqa: C901, PLR0915, PLR0912
         precision=32 if model == "boltz1" else "bf16-mixed",
     )
 
-    if filtered_manifest.records and not skip_structure:
+    if filtered_manifest.records:
         msg = f"Running structure prediction for {len(filtered_manifest.records)} input"
         msg += "s." if len(filtered_manifest.records) > 1 else "."
         click.echo(msg)

{boltz_vsynthes-1.0.2.dist-info → boltz_vsynthes-1.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: boltz-vsynthes
-Version: 1.0.2
+Version: 1.0.4
 Summary: Boltz for V-Synthes
 Requires-Python: <3.13,>=3.10
 Description-Content-Type: text/markdown

{boltz_vsynthes-1.0.2.dist-info → boltz_vsynthes-1.0.4.dist-info}/RECORD

@@ -1,5 +1,5 @@
 boltz/__init__.py,sha256=F_-so3S40iZrSZ89Ge4TS6aZqwWyZXq_H4AXGDlbA_g,187
-boltz/main.py,sha256=jesu0Y9aBVNPjulshsfyg78HTGNh2K9YjF76rWUWuFs,40833
+boltz/main.py,sha256=ZNaNplHLMr9OmLMhbivgducwbGCrn8g67WRPpuhh9GA,40084
 boltz/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 boltz/data/const.py,sha256=1M-88Z6HkfKY6MkNtqcj3b9P-oX9xEXluh3qM_u8dNU,26779
 boltz/data/mol.py,sha256=maOpPHEGX1VVXCIFY6pQNGF7gUBZPAfgSvuPf2QO1yc,34268
@@ -104,9 +104,9 @@ boltz/model/optim/scheduler.py,sha256=nB4jz0CZ4pR4n08LQngExL_pNycIdYI8AXVoHPnZWQ
 boltz/model/potentials/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 boltz/model/potentials/potentials.py,sha256=vev8Vjfs-ML1hyrdv_R8DynG4wSFahJ6nzPWp7CYQqw,17507
 boltz/model/potentials/schedules.py,sha256=m7XJjfuF9uTX3bR9VisXv1rvzJjxiD8PobXRpcBBu1c,968
-boltz_vsynthes-1.0.2.dist-info/licenses/LICENSE,sha256=8GZ_1eZsUeG6jdqgJJxtciWzADfgLEV4LY8sKUOsJhc,1102
-boltz_vsynthes-1.0.2.dist-info/METADATA,sha256=jjCm6Og0LpIvWLHgHJLnrHNkaGMO4cqZYf73yLKk8rM,7171
-boltz_vsynthes-1.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-boltz_vsynthes-1.0.2.dist-info/entry_points.txt,sha256=n5a5I35ntu9lmyr16oZgHPFY0b0YxjiixY7m7nbMTLc,41
-boltz_vsynthes-1.0.2.dist-info/top_level.txt,sha256=MgU3Jfb-ctWm07YGMts68PMjSh9v26D0gfG3dFRmVFA,6
-boltz_vsynthes-1.0.2.dist-info/RECORD,,
+boltz_vsynthes-1.0.4.dist-info/licenses/LICENSE,sha256=8GZ_1eZsUeG6jdqgJJxtciWzADfgLEV4LY8sKUOsJhc,1102
+boltz_vsynthes-1.0.4.dist-info/METADATA,sha256=5oK6_h_oPquPVFGAwT9rRFOd7F9RJ8jrC7UGKR8tRVQ,7171
+boltz_vsynthes-1.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+boltz_vsynthes-1.0.4.dist-info/entry_points.txt,sha256=n5a5I35ntu9lmyr16oZgHPFY0b0YxjiixY7m7nbMTLc,41
+boltz_vsynthes-1.0.4.dist-info/top_level.txt,sha256=MgU3Jfb-ctWm07YGMts68PMjSh9v26D0gfG3dFRmVFA,6
+boltz_vsynthes-1.0.4.dist-info/RECORD,,