pybiolib 1.2.980__tar.gz → 1.2.983__tar.gz

{pybiolib-1.2.980 → pybiolib-1.2.983}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pybiolib
-Version: 1.2.980
+Version: 1.2.983
 Summary: BioLib Python Client
 License: MIT
 Keywords: biolib

{pybiolib-1.2.980 → pybiolib-1.2.983}/biolib/_internal/utils/multinode.py

@@ -86,25 +86,22 @@ def fasta_batch_records(records, work_per_batch_min, work_per_residue=1, verbose
 
     batches = []
     batch = []
-    current_work_units = 0
-    total_work_units = 0
+    current_longest_seq_len = 0
     for record in records:
         # Add to batch
         batch.append(record)
 
-        # Calculate work units
-        seq = record.sequence
-        sequence_work_units = len(seq) * work_per_residue
+        seq_len = len(record.sequence)
+        current_longest_seq_len = max(current_longest_seq_len, seq_len)
 
-        # Increase counters
-        current_work_units += sequence_work_units
-        total_work_units += sequence_work_units
+        # Calculate work units based on longest sequence * number of sequences
+        current_work_units = current_longest_seq_len * work_per_residue * len(batch)
 
         # If above limit, start a new batch
         if current_work_units >= work_per_batch_min:
             batches.append(batch)
             batch = []
-            current_work_units = 0
+            current_longest_seq_len = 0
 
     # Append last batch if present
     if batch:

{pybiolib-1.2.980 → pybiolib-1.2.983}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pybiolib"
-version = "1.2.980"
+version = "1.2.983"
 description = "BioLib Python Client"
 readme = "PYPI_README.md"
 license = "MIT"