SURE-tools 2.1.40__tar.gz → 2.1.42__tar.gz

{sure_tools-2.1.40 → sure_tools-2.1.42}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SURE-tools
-Version: 2.1.40
+Version: 2.1.42
 Summary: Succinct Representation of Single Cells
 Home-page: https://github.com/ZengFLab/SURE
 Author: Feng Zeng

{sure_tools-2.1.40 → sure_tools-2.1.42}/SURE/perturb/perturb.py

@@ -117,15 +117,14 @@ def matrix_to_labels(matrix, unique_labels):
 def _numba_fill_matrix(dose_matrix, label_indices, label_doses):
     """Numba 加速的矩阵填充函数"""
     for i in range(len(label_indices)):
-        for j, dose in zip(label_indices[i], label_doses[i]):
-            dose_matrix[i, j] = dose
+        dose_matrix[i, label_indices[i]] = label_doses[i]
 
 def dose_to_matrix(labels, label_dose, all_labels=None):
     """
     使用 Numba 的终极加速版本（需预先安装 numba）
     """
     if all_labels is None:
-        all_labels = sorted(set().union(*labels))
+        all_labels = sorted(set().union(labels))
     
     label_to_idx = {label: idx for idx, label in enumerate(all_labels)}
     n_samples = len(labels)
@@ -135,16 +134,10 @@ def dose_to_matrix(labels, label_dose, all_labels=None):
     # 预处理为 Numba 兼容格式
     label_indices = []
     label_doses = []
-    for sample_labels in labels:
-        indices = []
-        doses = []
-        if sample_labels:
-            for label in sample_labels:
-                indices.append(label_to_idx[label])
-                doses.append(label_dose[label])
-        label_indices.append(np.array(indices, dtype=np.int64))
-        label_doses.append(np.array(doses, dtype=np.float64))
+    for i, label in enumerate(labels):
+        label_indices.append(label_to_idx[label])
+        label_doses.append(label_dose[i])
     
     # 调用 Numba 加速函数
     _numba_fill_matrix(dose_matrix, label_indices, label_doses)
-    return dose_matrix,all_labels
+    return dose_matrix,np.array(all_labels)

{sure_tools-2.1.40 → sure_tools-2.1.42}/SURE_tools.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SURE-tools
-Version: 2.1.40
+Version: 2.1.42
 Summary: Succinct Representation of Single Cells
 Home-page: https://github.com/ZengFLab/SURE
 Author: Feng Zeng

{sure_tools-2.1.40 → sure_tools-2.1.42}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setup(
     name='SURE-tools',
-    version='2.1.40',
+    version='2.1.42',
     description='Succinct Representation of Single Cells',
     long_description=long_description,
     long_description_content_type="text/markdown",